训练图像转换格式时出错。convert_data_to_tfrecord.py

Question

我想要达到的结果

D://FPN_TensorflowDNA/da/data_1010/0005.xml
Conversion progress:[                                        ]1%    1/598D://FPN_TensorflowDNA/da/data_1010/0006.xml
Traceback (most recent call last):
  File "D:/FPN_TensorflowDNA/data/io/convert_data_to_tfrecord.py", line 135, in <module>
    convert_pascal_to_tfrecord()
  File "D:/FPN_TensorflowDNA/data/io/convert_data_to_tfrecord.py", line 106, in convert_pascal_to_tfrecord
    img_height, img_width, gtbox_label = read_xml_gtbox_and_label(xml)
  File "D:/FPN_TensorflowDNA/data/io/convert_data_to_tfrecord.py", line 71, in read_xml_gtbox_and_label
    label = NAME_LABEL_MAP[child_item.text]
KeyError: 'Other'
Conversion progress:[                                        ]1%    2/598D://FPN_TensorflowDNA/da/data_1010/0007.xml

Process finished with exit code 1

训练图像转换时出错。提示出错的代码如上。
原来使用是没有问题的，新数据新问题，没找到原因。

Answer 1

这个应该是xml里面的label与代码中设置的label不对应，请检查一下xml中的label值other在代码中是否存在。有可能在代码中other label用了其他的表达方式