【RT-DETR实战】165、工业缺陷检测综合项目:模型改进与训练手记
一、深夜调参:一个诡异的收敛问题
上周三凌晨两点,盯着训练曲线发呆。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)<