Blade He

无法通过预训练模型完成训练

5

预训练模型路径： ./SegFormer/checkpoints/pretrained/mit_b1.pth 执行脚本： cd SegFormer python ./tools/train.py ./local_configs/segformer/B1/segformer.b1.512x512.ade.160k.py 报错信息： Traceback (most recent call last): File "c:\xxxx\github\mmcv\mmcv\utils\registry.py", line 51, in build_from_cfg return obj_cls(**args) File "C:\xxxx\github\torch_env\lib\site-packages\mmseg\models\segmentors\encoder_decoder.py", line 30, in __init__ self.backbone =...

执行脚本： python ./tools/test.py ./local_configs/segformer/B1/segformer.b1.512x512.ade.160k.py ./checkpoints/trained/segformer.b1.512x512.ade.160k.pth 报错信息： Traceback (most recent call last): File "c:\xxxx\github\mmcv\mmcv\utils\registry.py", line 51, in build_from_cfg return obj_cls(**args) File "C:\xxxx\github\torch_env\lib\site-packages\mmseg\datasets\pipelines\test_time_aug.py", line 59, in __init__ self.transforms = Compose(transforms) File "C:\xxxx\github\torch_env\lib\site-packages\mmseg\datasets\pipelines\compose.py",...