YOLO 模型训练踩坑日记

最近在做 YOLO 模型训练，遇到了不少问题，记录一下。

数据集格式

标注数据的时候一定要注意格式统一。之前有一批图片是 PNG 格式的，上传到标注平台后发现缩略图也是 PNG，体积直接翻了好几倍。后来统一转成 JPG 才解决。

显卡选择

RTX 4060 Laptop 训练 YOLOv11 足够用了，但要注意显存占用。batch size 开太大的话会 OOM，建议先用小 batch 跑通，再逐步调大。

模型量化

训练完的模型如果要在边缘设备上部署，量化是必经之路。ONNX 量化工具用起来还算方便，但要注意量化后的精度损失。

训练是个体力活，但看到模型指标一点点涨上去还是很有成就感的。