转发自https://blog.csdn.net/lgczym/article/details/79272579,原作者 Orankarl
为了使用预训练的TF-slim模型,下载了github上的TensorFlow/models库,随后运行research\object_detection\builders下的model_builder_test.py进行测试,提示ImportError: No module named 'nets'
进行搜索后发现一条相关的issue:ImportError: No module named nets,其中提到如下解决方法:
第一种解决方法:
添加环境变量PYTHONPATH,值为slim目录:然而该方法只能用于linux系统,Windows下不适用,即使添加环境变量也没起到效果。
第二种解决方法:
运行setup.py文件,将slim中所有的模块加载。
运行命令:
python setup.py build
python setup.py install
提示:
error: could not create 'build': 当文件已存在时,无法创建该文件。
原因是git clone下来的代码库中有个BUILD文件,而build和install指令需要新建build文件夹,名字冲突导致问题。暂时不清楚BUILD文件的作用。将该文件移动到其他目录,再运行上述指令,即可成功安装。随后重新运行python object_detection/builders/model_builder_test.py,结果如下:
Ran 7 tests in 0.167s------测试成功
参考链接:https://stackoverflow.com/questions/45036496/tensorflow-object-detection-importerror-no-module-named-nets