一、数据处理
将视频转成图片,并使用labelImg工具标注。
工具链接:https://github.com/tzutalin/labelImg
二、使用darknet训练模型
2.1 训练代码
使用darknet的代码进行训练,并添加了一些数据处理代码。详情可看readme以及代码注释。
yolov3:https://github.com/SeventhBlue/trainDarknet-yolov3
yolov4:https://github.com/SeventhBlue/trainDarknet-yolov4
2.2 部署代码
使用opencv440GPU版本进行部署。
opencv440GPU编译:https://blog.csdn.net/qq_31112205/article/details/105161419
yolov3和yolov4部署代码:https://github.com/SeventhBlue/opencvYolo
三、效果
yolov3衣物检测效果:https://www.bilibili.com/video/bv1Z54y167er
yolov4食物分拣效果:https://www.bilibili.com/video/bv1Fy4y187R1 |