深度学习-YOLOV3在VOC07数据集测试

本文分享了使用YOLOV3进行目标检测的训练过程,包括数据集介绍、训练策略及参数调整,以及最终的训练结果分析。特别指出在chair与pottedplant类别的检测效果提升,探讨了可能的原因。

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

github:https://github.com/wucj123/YOLOV3#yolov3

训练集

来自voc07,The PASCAL Visual Object Classification,此数据集分训练集与测试集,其中训练集,图片5011张,测试集图片4952张。

训练过程

去训练集中10%用作验证集,

阶段一:50个epoch,lr=1e-3,batch_size=32,耗时350s左右/epoch。

阶段二:100个epoch,lr=1e-4,batch_size=10,设置ReduceLROnPlateau,每3个epoch,val-loss如果不下降,则lr降为原来的0.1;设置EarlyStopping,如果10个epoch,val-loss不下降,则停止训练,保存weight。

训练结果

在val-loss降到25左右,提前停止训练。各类AP(iou=0.5)以及mAP如下所示

分析

分类效果比较差的类为chair与pottedplant,还在分析原因中,各位大神,知道原因的,欢迎评论,谢谢

è¿éåå¾çæè¿°

今天看到yolov2论文里面测试结果,其中chair与plant的AP分别为36.2与28.9,而yolov3中针对chair与plant,我目前测试结果为0.55与0.44,本身已有较大提升,所以原因可能在于数据集。

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值