目标检测：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 pycocotools

参考：https://blog.csdn.net/lilai619/article/details/106668913

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文件夹内

 

（7）训练

1）：数据格式转为coco格式的json文件

Tips:

1：图片名称必须为纯数字，且不要以0开头！！！

2：数据增广可以自己在getitem中自己做

 

可怜我的GPU是1060，显存就6G，只能跑BS=2, 输入512x512的图像。。。

 

 

 

 测试一下：效果一般，原因：增广不够好；只选择了D2结构；输入尺寸偏小。