Shaun ZhouandClaude Opus 5 43c8c4a179 fix(detect): 可视化推理自己按批切分，避免 CUDA OOM ...

model.predict 不理会传入的 batch 参数，把整个图片列表当成一个 batch，8969 张图在
16 GiB 显卡上直接 OOM（单次要分配 10.27 GiB）。改成自己按 --batch 切片调用。

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

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

..

build_yolo_dataset.py

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

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

check_env.py

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

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

cluster_unlabeled.py

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

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

export_clusters.py

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

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

generate_boxes.py

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

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

prepare_label_studio.py

feat(detect): 把 Label Studio 的 YOLO 导出装配成带 train/val 划分的数据集

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

visualize_detections.py

fix(detect): 可视化推理自己按批切分，避免 CUDA OOM

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