博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
VGG使用重复元素的网络
阅读量:6294 次
发布时间:2019-06-22

本文共 948 字,大约阅读时间需要 3 分钟。

由5个卷积层块(2个单卷积层,3个双卷积层),3个全连接层组成——VGG-11

from mxnet import gluon,init,nd,autogradfrom mxnet.gluon import nn,lossdef vgg_block(num_convs, num_channels):    blk = nn.Sequential()    for _ in range(num_convs):        blk.add(nn.Conv2D(num_channels,kernel_size=3,padding=1,activation='relu'))    blk.add(nn.MaxPool2D(pool_size=2,strides=2))    return blkconv_arch = ((1,64),(1,128),(2,256),(2,512),(2,512))def vgg(conv_arch):    net = nn.Sequential()    # 卷积层    for (num_convs,num_channels) in conv_arch:        net.add(vgg_block(num_channels,num_channels))    # 全连接层    net.add(nn.Dense(4096,activation='relu'),nn.Dropout(0.5),            nn.Dense(4096,activation='relu'),nn.Dropout(0.5),            nn.Dense(10)            )    return netnet = vgg(conv_arch)net.initialize()X = nd.random.uniform(shape=(1,1,224,224))for blk in net:    X = blk(X)    print(blk.name,'output shape:\t',X.shape)

输出形状。

 

转载于:https://www.cnblogs.com/TreeDream/p/10082108.html

你可能感兴趣的文章
2019测试指南-web应用程序安全测试(二)指纹Web服务器
查看>>
树莓派3链接wifi
查看>>
js面向对象编程
查看>>
Ruby中类 模块 单例方法 总结
查看>>
jQuery的validate插件
查看>>
5-4 8 管道符 作业控制 shell变量 环境变量配置
查看>>
Enumberable
查看>>
开发者论坛一周精粹(第五十四期) 求购备案服务号1枚!
查看>>
validate表单验证及自定义方法
查看>>
javascript 中出现missing ) after argument list的错误
查看>>
使用Swagger2构建强大的RESTful API文档(2)(二十三)
查看>>
Docker容器启动报WARNING: IPv4 forwarding is disabled. Networking will not work
查看>>
(转)第三方支付参与者
查看>>
程序员修炼之道读后感2
查看>>
DWR实现服务器向客户端推送消息
查看>>
js中forEach的用法
查看>>
Docker之功能汇总
查看>>
!!a标签和button按钮只允许点击一次,防止重复提交
查看>>
(轉貼) Eclipse + CDT + MinGW 安裝方法 (C/C++) (gcc) (g++) (OS) (Windows)
查看>>
还原数据库
查看>>