跑Boundary-Aware Feature Propagation遇到的问题

跑Boundary-Aware Feature Propagation遇到的坑

1、运行不报错,也不显示结果
从train.py代码第一行import开始添加,print(“check1”),发现问题torch.encoding无法导入,反复地参考github源代码https://github.com/zhanghang1989/PyTorch-Encoding重新安装。耗时一天,失败。给torch.encoding源码作者留言得到回复了,非常感谢~。
2、为还原作者使用的环境,AutoDL租了个2卡2080Ti,cuda 10.1,pytorch1.4 python3.6.2。
opencv版本高了安不上报错。
报错torch/lib/libtorch_python.so: undefined symbol: PySlice_Unpack, 把python安成3.6.2,好了。
pip版本低报错,upgrade一下就好了。
amazonaws.com权重下载不下来,从这下的res101权重,放到pretrainmoledel里。https://github.com/LikeLy-Journey/SegmenTron/releases/。
在这里插入图片描述
3、数据集格式不一样,重新写了个our.py
在这里插入图片描述

4、option.py文件添加断点续训
在这里插入图片描述
5、训练评估和测试指令
训练
python train.py --model bfp --dataset cityscapes --checkname bfp101 --batch-size 10 --workers 10
评估
python test.py --model bfp --dataset cityscapes --resume-dir cityscapes/bfp_model/bfp101_cropsize576_basesize608_batchsize10_lr0.0125_epoch100_UAG --eval
测试
python test.py --model bfp --dataset cityscapes --resume-dir cityscapes/bfp_model/bfp101_cropsize576_basesize608_batchsize10_lr0.0125_epoch100_UAG --test-folder datasets/test
测试生成图片在这里
在这里插入图片描述
在这里插入图片描述

n=2时,评估结果格式类似于:

0: 0.9448

1: 0.5906

Mean IoU over 2 classes: 0.7677

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

有梦想的炸豆皮

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值