我在使用 Tensorflow object_detection 的 python 项目中收到此错误

编程问答 2022-04-23

问题描述

创建 TF 记录

代码：

!python {SCRIPTS_PATH + '/generate_tfrecord.py'} -x {IMAGE_PATH + '/train'} -l {ANNOTATION_PATH + '/label_map.pbtxt'} -o {ANNOTATION_PATH + '/train.record'}

!python {SCRIPTS_PATH + '/generate_tfrecord.py'} -x{IMAGE_PATH + '/test'} -l {ANNOTATION_PATH + '/label_map.pbtxt'} -o {ANNOTATION_PATH + '/test.record'}

错误：

Traceback (most recent call last):
File "Tensorflow/scripts/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 "Tensorflow/scripts/generate_tfrecord.py",label_map_util
ModuleNotFoundError: No module named 'object_detection'

我关注了 Nicholas Renotte 的视频 https://youtu.be/pDXdlXlaCco @ 22:30

我从 git clone https://github.com/tensorflow/models

下载了 Tensorflow object_detection 模型

然后我跑了

python -m install pip

解决方法

之后 git clone https://github.com/tensorflow/models

您必须运行 protoc 才能正确安装 API：

cd models/research/
protoc object_detection/protos/*.proto --python_out=.
cp object_detection/packages/tf2/setup.py .
python -m pip install .

毕竟用这个脚本测试它： python object_detection/builders/model_builder_tf2_test.py

object-detection-apipippippythontensorflowtensorflowtensorflowtfrecord