我的新训练数据不断出现此错误。我已经尝试了示例数据,并且可以使用,但是当我使用自己的数据时,它给出了“ Key Error”。与我的数据和训练数据唯一不同的是,我的班级更多。 完全错误:
Traceback (most recent call last):
File "object_detection/create_tf_record.py", line 185, in <module>
tf.app.run()
File "C:\Users\edupt\AppData\Local\Programs\Python\Python36\lib\site-packages\tensorflow\python\platform\app.py", line 125, in run
_sys.exit(main(argv))
File "object_detection/create_tf_record.py", line 180, in main
image_dir, train_examples)
File "object_detection/create_tf_record.py", line 152, in create_tf_record
tf_example = dict_to_tf_example(data, label_map_dict, image_dir)
File "object_detection/create_tf_record.py", line 97, in dict_to_tf_example
classes.append(label_map_dict[class_name])
KeyError: '300424' <---------- THAT IS THE NAME OF ONE OF THE CLASSES