论文:[ECCV-2020]: Improving Semantic Segmentation via Decoupled Body and Edge Supervision
代码:https://github.com/lxtGH/DecoupleSegNets
不想用 ./scripts/train/train_cityscapes_ResNet50_deeplab.sh,直接改train.py
原来citiscapes是19类,我做的2分类,数据集按照citiscapes格式改好
记得改citiscapes.py,如下:
问题1:端口问题
train.py import一堆包之后加上
os.environ[‘MASTER_ADDR’] = ‘localhost’
os.environ[‘MASTER_PORT’] = ‘25649’
问题2:权重文件缺失
这俩权重都下载下来,第二个下载下来是".tar"结尾,但是解压失败,删掉“.tar”就行。
记得修改权重文件加载路径
resnet_d.py 315行
deepv3.py 244行
第一个epoch结果: