参考教程:https://blog.csdn.net/Layumi1993/article/details/84849506
https://zhuanlan.zhihu.com/p/50387521
一、需要安装的软件包
1.Python 3.6
2.Numpy
3.Pytorch 0.3+ (http://pytorch.org/)
方法一:
conda install torchvision -c pytorch
pip install torchvision
方法二:
Torchvision from the source
git clone https://github.com/pytorch/vision
cd vision
python setup.py install
额外补充包:
matplotlib
pretrainedmodels
ps:安装之前修复时间问题:
https://blog.csdn.net/jiuliang1916/article/details/80046977
二、训练
1.准备数据集 (python prepare.py)
那么现在打开刚刚下载的代码prepare.py。 将第五行的地址改为你本地的地址,比如 \home\zzd\Down