代码与模型

最近在做 YOLO 模型训练，遇到了不少问题，记录一下。 ## 数据集格式 标注数据的时候一定要注意格式统一。之前有一批图片是 PNG 格式的，上传到标注平台后发现缩略图也是 PNG，体积直接翻了好几倍。后来统一转成 JPG 才解决。 ## 显卡选择 RTX 4060 Laptop 训练 YOLOv11 足够用了，但要注意显存占用。batch size 开太大的话会 OOM，建议先用小 batch 跑通，再逐步调大。 ## 模型量化 训练完的模型如果要在边缘设备上部署，量化是必经之路。ONNX 量化工具用起来还算方便，但要注意量化后的精度损失。 --- 训练是个体力活，但看到模型指标一点点涨上去还是很有成就感的。

w w w w w ww w w w w w