key error in xml_style

[RiyazAina-DeepML]

i used your colab notebook and only changed the dataset inside your data folder.but when i ran train.py code i am getting following error ,dont know why?

KeyError: Caught KeyError in DataLoader worker process 0.
 File "/usr/local/lib/python3.6/dist-packages/mmdet-1.0+93bed07-py3.6-linux-x86_64.egg/mmdet/datasets/xml_style.py", line 47, in get_ann_info
    label = self.cat2label[name]
KeyError: 'agx'

[ahsan44411]

I have the same error. How did you fix it?

[CrazyCrud]

@RiyazAina-DeepML how many classes do you have in your dataset? I have the same problem with one class.