用CPU跑Deep-Residual-Shrinkage-Networks模型

用CPU跑Deep-Residual-Shrinkage-Networks模型

一、模型所在GitHub地址

https://github.com/zhao62/Deep-Residual-Shrinkage-Networks

二、环境配置

1. 软件环境

1. Python: Anaconda3
2. IDE: Pycharm

2. python库环境

1. tensorflow

   pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow==1.15
   

2. tflearn

   pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tflearn
   

三、代码运行

1. DRSN_TFLearn.py

1) 样本下载链接

因为代码自动下载速度太慢，所以直接手动下载

https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz

2) 代码和样本所在文件夹结构