老板需要做目标检测的项目于是学习了一下yolo的使用,在这里贴一些使用过程中的心得,方便日后查阅和交流
相关资源:
yolo官网:https://pjreddie.com/darknet/yolo/
yolo的pytorch实现:https://github.com/eriklindernoren/PyTorch-YOLOv3
针对pytorch实现yolo代码的注释:https://gitee.com/yeye0810/PyTorch-YOLOv3
图片标记工具:https://github.com/tzutalin/labelImg
yolo使用步骤:
相关参数修改
修改文件中的sets和class sets用于指定路径 class用于指定类别 根据voc_label.py的路径存放图片和xml文件 Main文件中的train.txt文件存放所有图片的名称(不带jpg)
cd darknet
cd cfg
vim yolov3-voc.cfg
文档开头需要指定训练格式还是test格式 由于自身计算能力的限制需要将batch更改为32(测试时出现存储转载的现象,所以将该batch改为32),更改yolo下的class和yolo上一层convolution的filter, filter的值为(cla