基于keras的YOLOV3改变锚框数量代码操作

本文介绍了如何将Keras版本的YOLOv3模型中原本的9个锚框调整为3个,详细步骤包括修改cfg文件、model.py文件,以及进行模型转换和训练。具体改动包括在cfg文件中设置num=3,mask和filters的调整,以及在model.py文件中修改num_layers和anchor_mask。
摘要由CSDN通过智能技术生成

起因

原来师兄的毕设里面有提到过改变锚框的数量,但是没有具体的实现流程,然后导师想要我试试看这个的效果,如果效果好可以放在项目里。于是就改yolo的代码。记录一下防止忘记。
注意:本文是将原有的9个anchors(每个尺度3个anchors)改为3个anchors(每个尺度1个anchors)。
基于keras版本的代码:https://github.com/qqwweee/keras-yolo3

步骤

1.修改cfg文件
找到yolov3.cfg,找到[yolo],总共有三个,此处我是留了原来每个尺度下面的中间锚框,如下所示16,30 , 62,45,156,198。然后修改num=3(原先为9),mask分别改为2,1,0.random和classes根据自身情况而定。改完后,每个[yolo]上面几行的filters改为x*(5+classes)其中X就是预设的anchor数量,此处我们预设的是1,所以总共为6.
在这里插入图片描述
2.修改model.py文件
num_layers要保证为3,所以在这个文件内出现的num_layer下面的//3,都给去掉。另外anchor_mask改为如下:

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值