KeyError: … is not in the dataset registry’

Published by onesixx on 22-02-2722-02-27

solution1>

recompile the code

pip install -v -e .

I thought I need to recompile the code

And I found that
for the newest version, the”DATASETS”is not imported from .registry.
So adding @DATASETS.register_module only is enough.

Also, for the newest version,
I build the mmdetection through pip install -v -e ..
So there is no need to reinstall after I add the new dataset

from .sixx_mydataset  import myDataset


__all__ = [
    'BaseDataset', .... 'CUB', 'CustomDataset',
    
    'myDataset',
]