如何解决我在使用 Tensorflow object_detection 的 python 项目中收到此错误
创建 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
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。