keras训练自己的yolo3模型

一、首先下载模型:https://download.csdn.net/download/OEMT_301/12505956
二、制作标签
三、配置训练模型
1、将原始图像拷贝到VOC2007/JPEGImages文件下
2、将制作的xml标签拷贝到VOC2007/ Annotations文件下
3、执行spilt.py在VOC2007\ImageSets\Main下生成4个txt文件
在这里插入图片描述
3、修改voc_annotation.py第6行,为自己的训练识别对象,我这里识别了两个物体分别是watch和person
在这里插入图片描述
4、修改model_data文件下的coco_classes.txt和voc_classes.txt为对应识别对象
在这里插入图片描述
在这里插入图片描述
5、执行voc_annotation.py可在dataset文件下得到三个txt文件
在这里插入图片描述
6、修改yolov3.cfg中第603、610、615行参数,与与之模型相对应
在这里插入图片描述
四、开始训练模型
1、在train.py第29行设置batch_size大小
在这里插入图片描述
2、在train.py第42行设置epochs次数
在这里插入图片描述
3、执行train.py,执行完成
4、执行完成会在checkpoint文件夹下生成trained_weights.h5文件
在这里插入图片描述
五、利用自己模型进行识别
1、将待识别图像拷贝到image文件下
2、执行yolo_video.py
3、当执行完毕,会在image\result文件下得到识别结果图像

在这里插入图片描述
我这里训练样本和次数都不够,因此结果较差。

参考:https://blog.csdn.net/Patrick_Lxc/article/details/80615433

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值