Receiving key error

[DhruvMakwana]

KeyError Traceback (most recent call last) in 1 if name == 'main': ----> 2 main()

in main() 28 for xml_file in os.listdir('../input/voc2012/VOC2007/Annotations'): 29 image_id = xml_file[:-4] ---> 30 if train_val_split[image_id] == 'train': 31 train_annotations.append(parse_one_xml(xml_file, names_map)) 32 elif train_val_split[image_id] == 'val':

KeyError: '008664'

how can I solve it?

[ethanyanjiali]

@DhruvMakwana it looks like the original dataset annotation is off

could you check if you have 008664 when you read the original annotation from this part of the code? I noticed that you put the dataset under ../input/voc2012, you'll need to do the same the code below too

with open('./VOCdevkit/VOC2012/ImageSets/Main/train.txt') as train_fp:
        lines = train_fp.read().splitlines()
        for line in lines:
            train_val_split[line] = 'train'