1 問題背景
TensorFlow Object Detection API 是以Slim為基礎實現的,需要將Slim的目錄加入PYTHONPATH后才能正確運行。
2 機器環境
window10
anaconda
3 解決辦法
Anaconda裝有多個環境,比如Base,TensorFlow-gpu,Pytorch等。
不同的環境,PYTHONPATH是不同的(很重要)。
假如我們想要在TensorFlow-gpu的環境中,將Slim的目錄加入其PYTHONPATH。方法如下:
- 首先激活TensorFlow-gpu的環境
activate TensorFlow-gpu
- 然后打印PYTHONPATH
python
import sys
print(sys.path)
- 查看Lib/site-packages目錄的具體位置。如下:
<img src="https://img2018.cnblogs.com/blog/1414369/201909/1414369-20190902152930510-827135695.jpg"width="440" align=center/>
- 創建一個.pth文件加到Lib/site-packages下即可。
注:
1).pth的文件內容為object detection 和slim的上一級目錄的位置(都在research目錄下)以及slim的位置:
D:\DeepCode\models\research
D:\DeepCode\models\research\slim
2)如何制作.pth文件?
新建一個文本文檔,將上述地址復制進去,然后把后綴名改為pth即可!!!
4 如何驗證加入到PYTHONPATH成功了?
以slim為例:
可以使用python命令打開一個python shell,如果運行import slim成功則說明已經正確設置好了。