awesome-semantic-segmentation-pytorch icon indicating copy to clipboard operation
awesome-semantic-segmentation-pytorch copied to clipboard

求助求助求助

Open lr1234567 opened this issue 4 years ago • 3 comments

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应该怎样改?

lr1234567 avatar Nov 02 '19 04:11 lr1234567

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

wuwenbin970731 avatar Dec 06 '19 13:12 wuwenbin970731

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

zjuxwg0401 avatar Apr 29 '20 14:04 zjuxwg0401

楼主问题解决了吗

Wongshy avatar Nov 07 '20 04:11 Wongshy