图像分割-----light weight refinenet

1.对于这个Model的训练,可以修改src/config.py的batch_size,类别数,是否需要pretrain_model,网络类型的选择等

2.对于test的时候,原作者给的是他自己的从网上下载的他自己训练好的Model,如果想用自己训练的model去做,需要修改对应load参数的程序,还有网络模型的选择。

3.不知道为什么自己训练的Model的测试结果和作者的差距比较大。自己训练作者的数据的时候,train_mean iou<0.5.分别为:原图、label、resnet101_selftrained_model

下图是使用作者的Model测试得到的结果,分别为:原图、label、resnet50、resnet101、resnet152的结果

4.而且训练的数据集的时候,mean iou>0.9,loss<0.1,但是测试结果很差,不是过拟合的问题,不知道是什么问题。

5.欢迎交流讨论

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值