search

Tony607 / mmdetection_object_detection_demo

How to train an object detection model with mmdetection
https://www.dlology.com/blog/how-to-train-an-object-detection-model-with-mmdetection/
Other
73 stars 194 forks source link

key error in xml_style #9

Open RiyazAina-DeepML opened 4 years ago

RiyazAina-DeepML commented 4 years ago

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 commented 4 years ago

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

CrazyCrud commented 3 years ago

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