AutoCNN:自定义CNN分类网络

代码下载地址:
- GitHub代码地址
- 代码百度网盘下载地址 提取码:8hzw
环境安装:
代码目录下,有个install.sh文件,运行:

bash install.sh

将会安装Pytorch、torchvision、opencv、numpy、tqdm的最新版本。
!!注意:这些或许会改变您的base环境。!!
建立模型:
AutoCNN的目的就是为了以后能更加有效地设计和使用CNN网络,避免麻烦地进行代码修改。
AutoCNN的网络结构是在model.conf文件中进行定义。如下为model_mnist.conf的网络结构:

conv:1,16,3,2,1
bn:16
ReLU
conv:16,32,3,2,1
bn:32
ReLU
conv:32,64,3,2,1
bn:64
ReLU
conv:64,64,2,2,0
bn:64
ReLU
fc:256,40

目前AutoCNN所支持的网络结构格式如下所示:

结构格式
卷积层(conv)conv:in_channel,out_channel,kernal_size,stride,padding
最大池化层(maxpool)maxpool:
平均池化层(avgpool)avgpool:
ReLU激活函数ReLU
sigmoid函数sigmoid
Dropout层dropout
BN层bn:input
全连接层(FC)fc:in_channel,out_channel
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值