参考教程: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\Download\Market,然后在终端中跑一下。
python prepa