我在带有 Tensorflow object_detection 的 python 项目中遇到此错误

Question

创建 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", line 29, in <module>
From object_detection.utils import dataset_util, 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 

Answer 1

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

Answer 2

如果您已经安装了相关的依赖项，例如 protoc 如下

%cd /content/models/research
!protoc object_detection/protos/*.proto --python_out=.

你需要创造环境，

!pip install tf_slim
pwd = os.getcwd()
os.environ['PYTHONPATH'] += f':{pwd}:{pwd}/slim'