跳到正文
duanyzhi
返回

ResNet

ResNet Paper

Github code for resnet

简单画了一个resnet101模型:

resnet

这里介绍一下: 首先这个模型的输入随机截取是[224, 224, 3]的ImageNet图片,如果是cifar10数据的话,模型是不太一样的。

我是在cifar10上跑的,ImageNet太慢了。cifar10的话Residual_block里面是两个卷积核大小是[3, 3]的卷积操作。也没有每一个conv之后的max pooling操作,是在每一个conv第一个卷积操作时将stride改成2来代替pooling,将特征图减小一半。

cifar10模型:

resnet_cifar10

当输入是cifar10是,卷积部分分为三块,总共卷积层有6n+2层。比如我是50层cnn,那么n=8. 一共分为三个大卷积块和一个全连接。所有的卷积核大小都是[3, 3]

resnet cifar10在实际些代码中发现需要一些技巧才能达到论文中准确率,我的Github上一个跑到0.93的模型:

resnet_cifar10_better

这里每一个residual block后面没有relu 并且不是每一个卷积都有bn。具体见github代码


分享这篇文章:

上一篇
Faster RCNN
下一篇
SHELL COMMANDS