接触过YOLO的朋友,应该都有过类似的经历:跑官方COCO示例顺风顺水,几行代码就能出检测效果,可一到自己做自定义场景,立马就卡壳。要么不知道怎么标注数据,要么标完格式不对,要么数据集路径配错,训练的时候各种玄学报错,本来以为半小时能搞定的事,折腾大半天还没跑起来。
我第一次做自定义数据集的时候,光格式转换和目录结构就踩了五六个坑,标注工具装了半小时装不上,训练的时候要么找不到标签,要么类别全对不上,来来回回返工,一下午就没了。后来做的项目多了,把整个流程标准化了一遍,从图片准备、标注、数据集划分到配置训练,整套走下来,熟练的话30分钟以内就能搞定一个小规模数据集,新手照着走也基本不会踩坑。
今天就把完整的实操步骤、核心脚本和踩过的坑全部分享出来,全程跟着做,就能从零跑出自己的YOLO检测模型。
一、前期准备:环境与工具搭建
不用一开始就装一堆复杂的依赖,核心就两个东西:标注工具和YOLO运行环境。
环境安装
推荐Python 3.8~3.11版本,太新的版本容易出现依赖兼容问题。两条命令搞定所有安装:
# 安装YOLO运行库,自带全部依赖pipinstallult