导入错误：没有名为“nets”的模块

Question

我试图从 tensorflow/models 中提供的 export_inference_graph.py 脚本将 trained_checkpoint 转换为最终冻结模型，但出现以下错误结果。是的，我已经将 $PYTHONPATH 设置为“models/slim”，但我仍然收到此错误，有人可以帮助我吗？

 $ echo $PYTHONPATH
:/home/ishara/tensorflow_models/models:/home/ishara/tensorflow_models/models/slim

\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*问题\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\* \*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\* ******

 $sudo python3 object_detection/export_inference_graph.py  --input_type image_tensor  --pipeline_config_path = "ssd_inception_v2_pets.config"  --trained_checkpoint_prefix="output/model.ckpt-78543"  --output_directory="birds_inference_graph.pb"

Traceback (most recent call last):
  File "object_detection/export_inference_graph.py", line 74, in <module>
    from object_detection import exporter
  File "/usr/local/lib/python3.5/dist-packages/object_detection-0.1-py3.5.egg/object_detection/exporter.py", line 28, in <module>

  File "/usr/local/lib/python3.5/dist-packages/object_detection-0.1-py3.5.egg/object_detection/builders/model_builder.py", line 30, in <module>
  File "/usr/local/lib/python3.5/dist-packages/object_detection-0.1-py3.5.egg/object_detection/models/faster_rcnn_inception_resnet_v2_feature_extractor.py", line 28, in <module>
ImportError: No module named 'nets'

---

几天来我一直在为此苦苦挣扎，尝试了很多解决方案都没有用我正在使用带有 tensorflow-gpu 版本的 Ubuntu 16.04。

原文由 Ishara Abeykoon 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

在 https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md 查看 Protobuf 编译并正确设置 PYTHONPATH，这就是我为 Windows 解决此问题的方法

对于 Windows：

来自张量流/模型/研究/

第一步： protoc object_detection/protos/*.proto --python_out=.

第2步：

 set PYTHONPATH= <Path to 'research' Directory> ; <Path to 'slim' Directory>

例如：

 set PYTHONPATH=C:\Users\Guy\Desktop\models\research;C:\Users\Guy\Desktop\models\research\slim

原文由 Pawan Mishra 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

Ubuntu 18 物理 GPU（设备：0，名称：Tesla K80，pci 总线 id：0000:00:1e.0，计算能力：3.7） 强文

cd models/research/slim/;
python setup.py build
python setup.py install

那是为了防止你下载或克隆了你的模型目录。

原文由 Vadim 发布，翻译遵循 CC BY-SA 4.0 许可协议