 Shaun ZhouandClaude Opus 5
|
154677032f
|
Revert "约束预测框互不重叠"（eba533a、9439cbb）
去掉强制非重叠：detect.py 不再给 DetectionValidator.postprocess 打补丁，
配置里移除 train.max_overlap 与 agnostic_nms，可视化脚本恢复为直接使用模型原始输出，
engine/detection 里的 suppress_overlaps / filter_overlapping_outputs 一并移除。
约束会把 mAP50 从 0.555 压到 0.483（precision 0.536->0.612），代价是按置信度抑制时
高分错框会压掉本可命中的低分框。需要时用 git revert 这次提交即可取回。
按类别分目录的可视化产出结构（36fff98）保留。
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
|
2026-09-18 15:34:13 +08:00 |
|
|
Shaun ZhouandClaude Opus 5
|
9439cbb56a
|
feat(detect): 训练期的验证也约束预测框互不重叠
ultralytics 没有可用的开关：yolo26 是端到端模型，non_max_suppression 对它早退
（utils/nms.py 里 end2end 分支），只按置信度与 max_det 截断，传入的 iou 和 agnostic
都不参与 IoU 抑制；agnostic_nms 只作用于检测头 top-k（每个 anchor 取一个类别），
压不掉不同 anchor 产生的重叠框。
所以给 DetectionValidator.postprocess 包一层 filter_overlapping_outputs。训练期验证、
早停、best.pt 的选择都随之遵守这条约束，指标与最终推理一致。配置项 train.max_overlap
（默认 0，负数关闭），由 detect.py 摘出后自行实现，不传给 ultralytics。
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
|
2026-09-18 11:30:48 +08:00 |
|
|
Shaun ZhouandClaude Opus 5
|
eba533a5b2
|
feat(detect): 约束预测框互不重叠
人工标注的真值里一处缺陷只有一个框：11211 张图、151 张多框图，重叠框对为 0。
而模型预测在验证集上有 6.9% 的图存在重叠框——NMS 只在同类内、按固定 IoU 阈值抑制，
压不掉两类情况：同一处缺陷被两个类别各框一次（实测 IoU 高达 0.99），以及 IoU 低于
NMS 阈值的同类重叠（grease 66 对、wrinkle 45 对）。
- engine/detection.suppress_overlaps：按置信度贪心保留互不重叠的框，可设允许的
最大 IoU；另可丢弃被大框完全包住的小框（面积悬殊时 IoU 很小，但显然是重复表达）。
- visualize_detections 默认 --max-overlap 0（与真值一致），并放出 --nms-iou 与
类别无关 NMS 开关；--max-overlap -1 可看模型原始输出。
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
|
2026-09-18 11:19:47 +08:00 |
|
|
Shaun ZhouandClaude Opus 5
|
788bc36ab8
|
feat(detect): 全量推理可视化，预测框与真值框画在一起
scripts/visualize_detections.py：在整个数据集上推理，绿=真值、蓝=命中的预测、
红=没对上真值的预测，按结果类型分目录（ok / missed / extra / none），再配一份
index.csv（逐图的真值数、预测数、命中数、最高置信度）与 summary.json（逐类 TP/FP/FN）。
--only errors 只输出有问题的图。
匹配逻辑放在 engine/detection.py：按置信度贪心、IoU 阈值、类别需一致，与 mAP 的规则
一致，所以图上看到的漏检误检和指标算的是同一回事。same_class=False 可以单看定位准不准
——定位对但分类错，和完全没找到是两种问题。
train 与 val 分开放：训练集的图模型见过，那部分结果不能当泛化能力看。
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
|
2026-09-17 21:47:02 +08:00 |
|