官方測試結果 :
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文件夾內