Tramac / awesome-semantic-segmentation-pytorch

Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
Apache License 2.0
2.85k stars 582 forks source link

求助求助求助 #98

Open lr1234567 opened 4 years ago

lr1234567 commented 4 years ago

Traceback (most recent call last): File "train.py", line 340, in trainer = Trainer(args) File "train.py", line 167, in init aux=args.aux, norm_layer=BatchNorm2d).to(self.device) File "/home/lr/桌面/awesome-semantic-segmentation-pytorch-master.2/core/models/model_zoo.py", line 137, in get_segmentation_model return modelsmodel File "/home/lr/桌面/awesome-semantic-segmentation-pytorch-master.2/core/models/deeplabv3_plus.py", line 131, in get_deeplabv3_plus model = DeepLabV3Plus(datasets[dataset].NUM_CLASS, backbone=backbone, pretrained_base=pretrained_base, kwargs) File "/home/lr/桌面/awesome-semantic-segmentation-pytorch-master.2/core/models/deeplabv3_plus.py", line 39, in init self.pretrained = get_xception(pretrained=pretrained_base, output_stride=output_stride, kwargs) File "/home/lr/桌面/awesome-semantic-segmentation-pytorch-master.2/core/models/base_models/xception.py", line 390, in get_xception model.load_state_dict(torch.load(get_model_file('xception', root=root))) File "/home/lr/桌面/awesome-semantic-segmentation-pytorch-master.2/core/models/model_store.py", line 68, in get_model_file raise ValueError('Model file is not found. Downloading or trainning.') ValueError: Model file is not found. Downloading or trainning.

deeplab v3 plus 用的基本模型xception,在哪找预训练模型,root应该怎样改?

wuwenbin970731 commented 4 years ago

在deeplabv3_plus.py中修改get_deeplabv3_plus函数的默认参数pretrained_base为false即可

zjuxwg0401 commented 4 years ago

我也遇到了相似的问题,请问楼主解决了吗?

Wongshy commented 3 years ago

楼主问题解决了吗