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.欢迎交流讨论