Windows下Tensorflow-slim库使用遇到ImportError:Nomodulenamednets问题的解决方法
| 分类: 机器视觉 |
为了使用预训练的TF-slim模型,下载了github上的TensorFlow/models库,随后运行models\research\slim\nets下的inception_resnet_v2_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文件夹,名字冲突导致问题。
需在models/research/目录下执行:
python setup.py install
可解决该问题。
如果后续出现:
ModuleNotFoundError: No module named 'pycocotools'
对于使用conda的,可以这样解决:
conda install pycocotools
如果出现:
ImportError: No module named 'nets'
需在models/research/slim目录下执行:
python setup.py install

加载中…