[我参加NVIDIA Sky Hackathon](模型训练ssd系列)

一、ssd-resnet18

比赛默认提供的目标检测模型是ssd,backbone是resnet18,默认训练80批次,部分训练参数及结果如下图1。

max map 0.8428856637198172
min train loss 4.685321
min val loss 44.12858346390398

图1 ssd-resnet18训练结果图 

二、ssd-resnet50

把resnet18改为resnet50,但是学习率没有调整好,导致过拟合了,实验结果如图二。

max map 0.48516015039328986
min train loss 8.325766
min val loss 74.27994304813751

图2 ssd-resnet50过拟合训练结果图 

调整学习率,提高epoch到300,map回归正常,实验结果如图3。

max map 0.8693470923063038
min train loss 5.8156857
min val loss 38.21444113468411

图3 ssd-resnet50训练结果图  

 三、ssd-resnet101

或许是参数没有调整好,相对于resnet18和resnet50效果并没有提升。

max map 0.8139915690690182
min train loss 17.634895
min val loss 49.02305871703696

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值