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

..

conftest.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_build_yolo_dataset.py

fix(detect): 钳掉 Label Studio 导出坐标的浮点溢出

2026-09-17 16:44:42 +08:00

test_classical.py

feat(boxes): 新增传统 CV 定位方法（--method cv），不依赖分类模型

2026-09-17 11:03:13 +08:00

test_classification.py

feat: 新增缺陷分类训练与评估链路

2026-09-10 14:58:41 +08:00

test_cluster.py

feat(cluster): 输出簇分布图，直观展示簇间距离

2026-09-04 15:09:35 +08:00

test_config.py

feat(detect): 默认模型换成 yolo26n

2026-09-11 14:34:16 +08:00

test_detection.py

feat(detect): 约束预测框互不重叠

2026-09-18 11:19:47 +08:00

test_device.py

feat: 支持 cuda / mps / cpu 三后端，Mac 上可用 MPS 训练

2026-09-02 21:43:35 +08:00

test_export_clusters.py

feat(cluster): clusters/ 默认改用硬链接，便于直接在 runs/ 下继续分子类

2026-09-10 12:04:08 +08:00

test_generate_boxes.py

feat(boxes): 新增传统 CV 定位方法（--method cv），不依赖分类模型

2026-09-17 11:03:13 +08:00

test_gradcam.py

feat(localization): SHAP Partition 定位方法，并把每图框数上限默认收紧为 1

2026-09-10 16:10:50 +08:00

test_models.py

refactor: 按实际的五步工作流裁掉无关代码

2026-09-11 10:27:20 +08:00

test_prepare_label_studio.py

fix(label-studio): 图片 URL 按实际的 DOCUMENT_ROOT 生成（--document-root）

2026-09-15 20:04:04 +08:00

test_registry.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_shap_explain.py

feat(localization): SHAP Partition 定位方法，并把每图框数上限默认收紧为 1

2026-09-10 16:10:50 +08:00

test_simclr.py

feat(cluster): clusters/ 默认改用硬链接，便于直接在 runs/ 下继续分子类

2026-09-10 12:04:08 +08:00

test_unlabeled_dataset.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00

test_visualize_detections.py

feat(detect): 全量推理可视化，预测框与真值框画在一起

2026-09-17 21:47:02 +08:00

test_visualizer.py

test: 单元测试与端到端冒烟

2026-09-02 17:55:47 +08:00