0

!python {'/content/generate_tfrecord.py'} -x {'/content/Training'} -l {'/content/label_map.pbtxt'} -o {ANNOTATION_PATH + '/train.record'}
!python {'/content/generate_tfrecord.py'} -x{'/content/Testing'} -l {'/content/label_map.pbtxt'} -o {ANNOTATION_PATH + '/test.record'}

运行这给了我错误:

Traceback (most recent call last):
  File "/content/generate_tfrecord.py", line 29, in <module>
    from object_detection.utils import dataset_util, label_map_util
ModuleNotFoundError: No module named 'object_detection'
Traceback (most recent call last):
  File "/content/generate_tfrecord.py", line 29, in <module>
    from object_detection.utils import dataset_util, label_map_util
ModuleNotFoundError: No module named 'object_detection'

MacOS Catalina 10.15.2,Tensorflow(最新版本)

我已经通过 pip 安装了所有依赖项。(object-detection api,在终端中导出路径,运行命令“python setup.py install in the same path)

谢谢!

4

1 回答 1

1

您已经正确编译了 proto 文件。

export PYTHONPATH=$PYTHONPATH:<PATH_TO_TF>/TensorFlow/models/research
export PYTHONPATH=$PYTHONPATH:<PATH_TO_TF>/TensorFlow/models/research/object_detection
export PYTHONPATH=$PYTHONPATH:<PATH_TO_TF>/TensorFlow/models/research/slim

导出后,在里面tensorflow/models/research就可以运行了

python setup.py build
python setup.py install

它会起作用的。我之前在我的mac os上测试过。

于 2021-01-12T11:36:15.440 回答