Windows+Pytorch+EfficientDet

官方測試結果 :

 

zylo開源測試結果: 

win7+python3.7+cuda10.1+pytorch1.4

(1)安裝anaconda【先查看py37對應的anaconda版本】

https://repo.anaconda.com/archive/

https://docs.anaconda.com/anaconda/packages/oldpkglists/

我這裏是py37,anaconda就是Anaconda3-5.3.0-Windows-x86_64.exe,默認安裝即可。

 

(2)安裝ffmpeg【torchvision會用到】

不安裝,可能會提示:torchvision installation problem: avcodec-58.dll not found

1):https://ffmpeg.zeranoe.com/builds/ 下載對應版本的動態庫

2):將bin文件添加到系統環境變量

 

(3)安裝mkl版本的numpy

不安裝,可能會提示:from scipy.linalg import _fblas    

1):到https://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy,

2):根據自己的python版本和numpy版本找到對應的文件下載對應的whl文件

3):pip install 此安裝文件   

我這裏是:numpy-1.16.4+mkl-cp37-cp37m-win_amd64.whl

上面的網站下載太慢了,可以到這裏下載:https://download.csdn.net/download/lilai619/12514864

 

我這裏還提示了:

twisted 18.7.0 requires PyHamcrest>=1.9.0, which is not installed.

直接安裝就可以了:

pip install -i https://mirrors.aliyun.com/pypi/simple/ PyHamcrest==1.9.0

 

(4)安裝pytorch, torchvision

1):離線下載torch,torchvision包

https://download.pytorch.org/whl/torch_stable.html

 

 

2):使用pip install安裝   

pip install torch-1.4.0-cp37-cp37m-win_amd64.whl
pip install torchvision-0.5.0-cp37-cp37m-win_amd64.whl

 

(5)剩下的缺少裝啥就行了

1):no module named cv2

pip install opencv-python

2):no module named albumentations

pip install albumentations

3):no module named tensorboard tensorboardX

pip install tensorboard
pip install tensorboardX

3):no module named webcolors

pip install webcolors

 

(6)安裝EfficientDet

1):下載 https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch

2):下載權重

 

3):創建weights文件夾,將權重文件移動到該文件夾

4):運行efficientdet_test.py,測試圖會生成在test文件夾內

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章