使用darknet训练yolov3、yolov4模型并部署:食物分拣和衣物穿戴检测

论坛 期权论坛 编程之家     
选择匿名的用户   2021-5-28 15:21   11   0

一、数据处理

将视频转成图片,并使用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

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP