如何在制作tfrecord文件时解决错误

Question

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

'''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

Answer 1

generate_tfrrecord.py找不到对象检测实用程序的位置。

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

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

或

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

  sys.path.append('/Users/path to tensorflow code/models/research'(

(填写设备上tensorflow代码所在位置的正确路径(