将 xml 文件转换为 tfrecord 文件时出错答案

【问题标题】：Error when converting xml files to tfrecord files将 xml 文件转换为 tfrecord 文件时出错
【发布时间】：2020-12-25 01:30:34
【问题描述】：

我正在关注 Macbook 上的 TensorFlow 2 对象检测 API 教程

这是我在运行将 xmls 转换为 TFrecords 的给定脚本时得到的结果

Traceback (most recent call last):
  File "generate_tfrecord.py", line 62, in <module>
    label_map_dict = label_map_util.get_label_map_dict(label_map)
  File "/usr/local/lib/python3.8/site-packages/object_detection/utils/label_map_util.py", line 164, in get_label_map_dict
    label_map = load_labelmap(label_map_path)
  File "/usr/local/lib/python3.8/site-packages/object_detection/utils/label_map_util.py", line 133, in load_labelmap
    label_map_string = fid.read()
  File "/usr/local/lib/python3.8/site-packages/tensorflow/python/lib/io/file_io.py", line 116, in read
    self._preread_check()
  File "/usr/local/lib/python3.8/site-packages/tensorflow/python/lib/io/file_io.py", line 78, in _preread_check
    self._read_buf = _pywrap_file_io.BufferedInputStream(
TypeError: __init__(): incompatible constructor arguments. The following argument types are supported:
    1. tensorflow.python._pywrap_file_io.BufferedInputStream(arg0: str, arg1: int)

Invoked with: item {
  name: "cat"
  id: 1
}
, 524288

我的标签映射文件包含以下内容

item {
    id: 1
    name: 'cat'
}

【问题讨论】：

你能在你的问题中添加你的标签映射文件吗？
@YuexinXu 请将 beryllite 的回答标记为已接受。

标签： python tensorflow object-detection

【解决方案1】：

看来问题可以通过更换解决

label_map = label_map_util.load_labelmap(args.labels_path)
label_map_dict = label_map_util.get_label_map_dict(label_map)

作为

label_map_dict = label_map_util.get_label_map_dict(args.labels_path)

【讨论】：

这对我也有用。您能解释一下为什么其他版本不起作用吗？
谢谢，它不仅适用于 mac，还解决了我在 windows 上的错误。

【解决方案2】：

我遇到了同样的错误并找到了解决方法。

删除线：

label_map = label_map_util.load_labelmap(args.labels_path)
label_map_dict = label_map_util.get_label_map_dict(label_map)

并根据您的标签映射更改 def class_text_to_int，如下所示：

def class_text_to_int(row_label):
    if row_label == 'cat':
        return 1

现在一切都应该正常了。

【讨论】：

【解决方案3】：

确保从 github 上的 tensorflow 目录克隆对象检测文件： https://github.com/tensorflow/models

我从不同的github目录克隆了这些文件，然后意识到它们不一样，其他目录有很多问题

【讨论】：