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应该怎样改?
在deeplabv3_plus.py中修改get_deeplabv3_plus函数的默认参数pretrained_base为false即可