【RT-DETR实战】170、遥感目标检测综合项目:UCAS-AOD数据集实战
昨天深夜调一个模型,输出层突然报shape不匹配。
debug了半小时才发现,UCAS-AOD里那张该死的“1320.tif”图像通道数居然是1——灰度图混在RGB数据集里,forward到一半崩了。
做遥感项目就是这样,你以为只是换套数据跑检测,实际从数据加载开始就得重新造轮子。
数据这关就够喝一壶
UCAS-AOD这数据集,飞机和车辆两类目标,分辨率从0.5米到2米都有。官方给的标注是.mat文件,MATLAB格式。
第一次见的时候我还想偷懒转成VOC格式,结果发现坐标是四个点(x1,y1,x2,y2,x3,y3,x4,y4),不是常规的矩形框。
得自己算最小外接矩形,这里踩过坑:直接取min/max会丢角度信息,遥感目标可都是带旋转的。
# 别这样写,会丢旋转信息x_min=min(x1,x2,x3,x4)y_min