I have removed cache from the VOCdevkit folder,and the pascal_voc.py have modified to my own label.
self.classes = ['background', #always index 0
'DiaoChe','TaDiao', 'ShiGongJiXie', 'DaoXianYiWu', 'YanHuo']
When I run my data set, I always report an error:
Traceback (most recent call last):
File "main_train.py", line 72, in
for image_set in image_sets]
File "lib/data_utils/load_data.py", line 29, in load_proposal_roidb
roidb = imdb.gt_roidb()
File "lib/dataset/pascal_voc.py", line 105, in gt_roidb
gt_roidb = [self.load_pascal_annotation(index) for index in self.image_set_index]
File "lib/dataset/pascal_voc.py", line 168, in load_pascal_annotation
cls = class_to_index[obj.find('name').text.lower().strip()]
KeyError: 'yanhuo'
Can you give me a hand?
I have removed cache from the VOCdevkit folder,and the pascal_voc.py have modified to my own label. self.classes = ['background', #always index 0 'DiaoChe','TaDiao', 'ShiGongJiXie', 'DaoXianYiWu', 'YanHuo'] When I run my data set, I always report an error: Traceback (most recent call last): File "main_train.py", line 72, in
for image_set in image_sets]
File "lib/data_utils/load_data.py", line 29, in load_proposal_roidb
roidb = imdb.gt_roidb()
File "lib/dataset/pascal_voc.py", line 105, in gt_roidb
gt_roidb = [self.load_pascal_annotation(index) for index in self.image_set_index]
File "lib/dataset/pascal_voc.py", line 168, in load_pascal_annotation
cls = class_to_index[obj.find('name').text.lower().strip()]
KeyError: 'yanhuo'
Can you give me a hand?