时间戳

首页 > 解决方案 > 制作 tfrecord 文件时如何解决错误？

python - 制作 tfrecord 文件时如何解决错误？

问题描述

所以伙计们，我遇到了这个脚本的错误

'''python generate_tfrecord.py --csv_input train_labels.csv --image_dir train --labelmap_dir label_map.pbtxt --output_path train.record'''

然后评论是这样的：

Traceback (most recent call last):
  File "generate_tfrecord.py", line 15, in <module>
    from object_detection.utils import dataset_util
ImportError: No module named object_detection.utils

标签： pythontensorflowobject-detectionobject-detection-apitfrecord

解决方案

您的 generate_tfrecord.py 找不到您的对象检测工具位置。

您需要克隆/下载 tensorflow 对象检测代码，然后您有 2 个选项：

  • 将文件放在研究文件夹中并在那里运行

或者

  • 在 python 文件的顶部添加以下行：

    sys.path.append('/Users/ tensorflow 代码路径 /models/research')

（填写正确的路径到您的设备上tensorflow代码所在的位置）

推荐阅读