Win10+keras的YOLOV3的复现+自己的数据集

本文详细介绍了在Windows 10系统中使用Keras复现YOLOV3模型的过程,包括代码下载、权重转换、模型训练、数据集制作、类别调整以及模型测试。通过训练自定义数据集,调整模型参数,最终得到识别效果提升的模型。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1、代码下载:下载yolov3代码:https://github.com/qqwweee/keras-yolo3 ,并解压缩之后用pycharm打开。

2、下载权重:https://pjreddie.com/media/files/yolov3.weights

3、生成模型:将darknet下的yolov3配置文件转换成keras适用的h5文件:

python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5

4、运行预测单张图像程序

python yolo_video.py --image

5、训练VOC2007,可以将自己的数据集制作成VOC2007格式:

放置:D:\demo\keras-yolo3-master\VOCdevkit\VOC2007\四个标准的文件夹

 


6、在voc_annotation.py需改你的数据集的类别:修改成自己的类

python voc_annotation.py

生成3个文件:三个新的不同的txt文件:2007_train.txt; 2007_test.txt; 2007_val.txt

7、运行Kmeans.py生成新的锚文件yolo_anchors.txt,并将其复制到model_data文件夹下

8、修改yolo3.cfg文件:

 filter:3*(5+len(classes))我这里只有一个类,所以filter=18
                 
 classes:你要训练的类别数(我这里是训练1类) 
                  
 r

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值