AttributeError: 'EncoderDecoder' object has no attribute 'CLASSES'

[Htermotto]

Hi,

When trying to implement a new backbone into a segmentation class and run the inference on a single image as per the example notebook I get this error

AttributeError: 'EncoderDecoder' object has no attribute 'CLASSES'.

I believe this stems somehow from the BaseModule in mmcv not having it's classes set from the ../_base_/data/cityscapes.py folder, but I'm not sure how to fix this, or how this attribute CLASSES gets set in the EncoderDecoders that work. Some help or pointers to documentation would be appreciated.

EDIT: I'm including the link to the cityscapes.py base in my config file which then gets passed to init_segmentor similar to the other models.

Thanks!

[Htermotto]

Ah this needs to be set manually? I was able to get around my issues after doing so.

I referenced: https://github.com/open-mmlab/mmsegmentation/blob/master/tools/train.py#L226

[MengzhangLI]

It is wired because I think you do not need to set classes of dataset manually if you have inherited cityscapes dataset config. It has been included in its dataset class. Could you debug whether dataset config parameters have been imported?

[Htermotto]

Thanks for getting back to me so quickly!

What's the best way to see if whether the dataset config parameters have been imported? Where would I check that?

[MengzhangLI]

Thanks for getting back to me so quickly!

What's the best way to see if whether the dataset config parameters have been imported? Where would I check that?

You can use PyCharm or VSCode to debug. Or check out log (in training) where config parameters are all recorded.