Pycharm上运行测试文件出错ImportError: cannot import name 'anchor_generator_pb2'

Question

问题遇到的现象和发生背景

运行

VERIFICATION_SCRIPT = os.path.join(paths['APIMODEL_PATH'], 'research', 'object_detection', 'builders', 'model_builder_tf2_test.py')
# Verify Installation
!python {VERIFICATION_SCRIPT}

测试文件
报错

问题相关代码，请勿粘贴截图

VERIFICATION_SCRIPT = os.path.join(paths['APIMODEL_PATH'], 'research', 'object_detection', 'builders', 'model_builder_tf2_test.py')
# Verify Installation
!python {VERIFICATION_SCRIPT}

运行结果及报错内容

ImportError: cannot import name 'anchor_generator_pb2' from 'object_detection.protos' (C:\Users\ZZH\PycharmProjects\tfnew\venv\lib\site-packages\object_detection\protos_init_.py)

我的解答思路和尝试过的方法

都重头再来了没辙了

我想要达到的结果

正常运行测试文件

Answer 1

原因
在执行这个之前，你需要先编译一下proto文件，可能你没有执行这一步

protoc object_detection/protos/*.proto --python_out=.

如果还不行的话，那就是环境变量的问题
尝试下面的解决方法。

---

解决步骤：

方法一
1、正常安装
2、添加环境变量，可参考：

Windows7下编译 tensorflow 目标检测api运行产生：cannot import name 'anchor_generator_pb2'_Winter-Solider的博客-CSDN博客 初次运行过程中遇到的问题：cannot import name 'anchor_generator_pb2'解决上一个问题过程中遇到的问题：object_detection/protos/*.proto: No such file or directory第一个问题：models\research\object_detection\protos 路径下的 .proto 文件需要编译成 .py 文件... https://blog.csdn.net/qq_22910255/article/details/80516554

3、在执行之前model_builder_test.py，您需要编译 proto 文件，如下所示：

# From tensorflow/models/
protoc object_detection/protos/*.proto --python_out=.

方法二
对于windows来说 可以用源码安装的简单快速的方法

1、为 Windows 下载的 protobuf v3.4：https://github.com/google/protobuf/releases/tag/v3.4.0
2、将 protoc.exe 放在“research”文件夹中
3、然后运行下面的命令。

protoc object_detection/protos/.proto --python_out=.

python object_detection/builders/model_builder_test.py

anchor_generator_pb2.py 文件位于 object_detection/protos

参考链接：

[Solved] models cannot import name 'anchor_generator_pb2' I ran into the same error Steps to resolve:Read the Installation Instruction hereBefore executing model_builder_test.py https://lifesaver.codes/answer/cannot-import-name-anchor-generator-pb2-1962

如果上述方法还不行，可以参考：

Windows7下编译 tensorflow 目标检测api运行产生：cannot import name 'anchor_generator_pb2'_Winter-Solider的博客-CSDN博客 初次运行过程中遇到的问题：cannot import name 'anchor_generator_pb2'解决上一个问题过程中遇到的问题：object_detection/protos/*.proto: No such file or directory第一个问题：models\research\object_detection\protos 路径下的 .proto 文件需要编译成 .py 文件... https://blog.csdn.net/qq_22910255/article/details/80516554

---

如有问题及时沟通