用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) 代码和样本所在文件夹结构

在这里插入图片描述

3) 运行结果

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

2. DRSN_Keras.py

1) 代码修改

  1. 需修改的代码

    import keras
    from keras.datasets import mnist
    from keras.layers import Dense, Conv2D, BatchNormalization, Activation
    from keras.layers import AveragePooling2D, Input, GlobalAveragePooling2D
    from keras.optimizers import Adam
    from keras.regularizers import l2
    from keras import backend as K
    from keras.models import Model
    from keras.layers.core import Lambda
    
  2. 修改后代码

    from tensorflow import keras
    from tensorflow.keras.datasets import mnist
    from tensorflow.keras.layers import Dense, Conv2D, BatchNormalization, Activation
    from tensorflow.keras.layers import AveragePooling2D, Input, GlobalAveragePooling2D 
    from tensorflow.keras.optimizers import Adam
    from tensorflow.keras.regularizers import l2
    from tensorflow.keras import backend as K
    from tensorflow.keras.models import Model
    from tensorflow_core.python.keras.layers import Lambda
    

2)运行结果

在这里插入图片描述

在这里插入图片描述

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
©️2022 CSDN 皮肤主题:1024 设计师:我叫白小胖 返回首页
评论

打赏作者

Computer-Rookie

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值