一、深夜调参:一个诡异的收敛问题
上周三凌晨两点,盯着训练曲线发呆。Loss在epoch 10之后就开始跳舞——时而下降,时而突然飙升,验证集mAP卡在0.72死活上不去。
工业缺陷检测的数据集有个特点:正负样本极度不均衡,一张钢板图像上可能只有两三个微小划痕,背景却占99%的面积。
RT-DETR原版的匹配策略在这里明显水土不服。
# 原版匈牙利匹配直接用的代价矩阵
cost_matrix = self.cls_cost(pred_logits, gt_labels) + \\
self.box_cost(pred_boxes, gt_boxes)<
