yolov3+keras之模型和权重加载

本文介绍了如何使用Keras实现YOLOv3目标检测模型,并详细阐述了加载预训练权重的过程,包括按名称对应加载权重、设置冻结层的两种模式以及IOU(交并比)的计算方法。
摘要由CSDN通过智能技术生成

https://blog.csdn.net/weixin_33743703/article/details/87985568

https://github.com/SpikeKing/keras-yolo3-detection

实现代码 keras版本

https://github.com/qqwweee/keras-yolo3

训练时的网络模型

def create_model(input_shape, anchors, num_classes, load_pretrained=True, freeze_body=2,
                 weights_path='model_data/yolo_weights.h5'):
#input_shape:输入图片的尺寸,默认是(416, 416);
#anchors:默认的9种anchor box,结构是(9, 2);
#num_classes:类别个数,在创建网络时,只需类别数即可。在网络中,类别值按0~n排列,同时,输入数据的类别也是用索引表示;
#load_pretrained:是否使用预训练权重。预训练权重,既可以产生更好的效果,也可以加快模型的训练速度;
#freeze_body:冻结模式,1或2。其中,1是冻结DarkNet
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值