Error in visualization of CWD test results.

[sunshiding]

I get an error when I execute the following command:

python tools/mmseg/test_mmseg.py /home/sunshiding/mmrazor/work_dirs/cwd_cls_head_pspnet_r50_d8_pspnet_r18_d8_512x512_voc12/cwd_cls_head_pspnet_r50_d8_pspnet_r18_d8_512x512_voc12.py /home/sunshiding/mmrazor/work_dirs/cwd_cls_head_pspnet_r50_d8_pspnet_r18_d8_512x512_voc12/iter_4000.pth --show-dir ./results/cwd_cls_head_pspnet_r50_d8_pspnet_r18_d8_512x512_voc12

The error is as follows：

load checkpoint from local path: /home/sunshiding/mmrazor/work_dirs/cwd_cls_head_pspnet_r50_d8_pspnet_r18_d8_512x512_voc12/iter_4000.pth
[ ] 0/115, elapsed: 0s, ETA:Traceback (most recent call last):
File "tools/mmseg/test_mmseg.py", line 248, in
main()
File "tools/mmseg/test_mmseg.py", line 210, in main
format_args=eval_kwargs)
File "/home/sunshiding/mmsegmentation-master/mmseg/apis/test.py", line 140, in single_gpu_test
opacity=opacity)
File "/home/sunshiding/.conda/envs/razor/lib/python3.7/site-packages/mmrazor/models/algorithms/base.py", line 170, in show_result
return self.architecture.show_result(img, result, **kwargs)
File "/home/sunshiding/.conda/envs/razor/lib/python3.7/site-packages/mmrazor/models/architectures/base.py", line 41, in show_result
return self.model.show_result(img, result, **kwargs)
File "/home/sunshiding/mmsegmentation-master/mmseg/models/segmentors/base.py", line 242, in show_result
assert palette.shape[0] == len(self.CLASSES)
File "/home/sunshiding/.conda/envs/razor/lib/python3.7/site-packages/torch/nn/modules/module.py", line 779, in getattr
type(self).name, name))
torch.nn.modules.module.ModuleAttributeError: 'EncoderDecoder' object has no attribute 'CLASSES'

[wwjwy]

you can change File "/home/sunshiding/mmsegmentation-master/mmseg/models/segmentors/base.py", line 242, in show_result
assert palette.shape[0] == len(self.CLASSES) to assert palette.shape[0] == 21(the number of classes), i meet the same problem and solve it like this.

[wutongshenqiu]

Thanks for your issue. This is probably due to the fact that the CLASSES property is only assigned to algorithm instead of the model used in mmsegmentation, we will fix this soon.

[wutongshenqiu]

Could you please modify test_mmseg.py according to this pr open-mmlab/mmrazor#53 and see if there is still an error?

[sunshiding]

Could you please modify test_mmseg.py according to this pr open-mmlab/mmrazor#53 and see if there is still an error?

Thank you very much, this problem has been solved perfectly.