Skip to content

Commit

Permalink
Revise the zero paddings of pool1 in ResNets
Browse files Browse the repository at this point in the history
  • Loading branch information
@taehoonlee
taehoonlee committed Sep 1, 2018
1 parent 6fc2111 commit c3b5320
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
6 changes: 3 additions & 3 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -260,9 +260,9 @@ with tf.Session() as sess:
| [ResNet50](tensornets/resnets.py#L85) | 25.126 | 7.982 | 6.842 | 25.6M | 23.6M | 195.4 | [[paper]](https://arxiv.org/abs/1512.03385) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v1.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/resnet.lua) <br /> [[caffe]](https://github.com/KaimingHe/deep-residual-networks/blob/master/prototxt/ResNet-50-deploy.prototxt) [[keras]](https://github.com/keras-team/keras/blob/master/keras/applications/resnet50.py) |
| [ResNet101](tensornets/resnets.py#L113) | 23.580 | 7.214 | 6.092 | 44.7M | 42.7M | 311.7 | [[paper]](https://arxiv.org/abs/1512.03385) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v1.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/resnet.lua) <br /> [[caffe]](https://github.com/KaimingHe/deep-residual-networks/blob/master/prototxt/ResNet-101-deploy.prototxt) |
| [ResNet152](tensornets/resnets.py#L141) | 23.396 | 6.882 | 5.908 | 60.4M | 58.4M | 439.1 | [[paper]](https://arxiv.org/abs/1512.03385) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v1.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/resnet.lua) <br /> [[caffe]](https://github.com/KaimingHe/deep-residual-networks/blob/master/prototxt/ResNet-152-deploy.prototxt) |
| [ResNet50v2](tensornets/resnets.py#L98) | 24.526 | 7.252 | 6.012 | 25.6M | 23.6M | 209.7 | [[paper]](https://arxiv.org/abs/1603.05027) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/preresnet.lua) |
| [ResNet101v2](tensornets/resnets.py#L126) | 23.116 | 6.488 | 5.230 | 44.7M | 42.6M | 326.2 | [[paper]](https://arxiv.org/abs/1603.05027) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/preresnet.lua) |
| [ResNet152v2](tensornets/resnets.py#L154) | 22.236 | 6.080 | 4.960 | 60.4M | 58.3M | 455.2 | [[paper]](https://arxiv.org/abs/1603.05027) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/preresnet.lua) |
| [ResNet50v2](tensornets/resnets.py#L98) | 24.040 | 6.966 | 5.896 | 25.6M | 23.6M | 209.7 | [[paper]](https://arxiv.org/abs/1603.05027) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/preresnet.lua) |
| [ResNet101v2](tensornets/resnets.py#L126) | 22.766 | 6.184 | 5.158 | 44.7M | 42.6M | 326.2 | [[paper]](https://arxiv.org/abs/1603.05027) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/preresnet.lua) |
| [ResNet152v2](tensornets/resnets.py#L154) | 21.968 | 5.838 | 4.900 | 60.4M | 58.3M | 455.2 | [[paper]](https://arxiv.org/abs/1603.05027) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/preresnet.lua) |
| [ResNet200v2](tensornets/resnets.py#L169) | 21.714 | 5.848 | 4.830 | 64.9M | 62.9M | 618.3 | [[paper]](https://arxiv.org/abs/1603.05027) [[tf-slim]](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py) [[torch-fb]](https://github.com/facebook/fb.resnet.torch/blob/master/models/preresnet.lua) |
| [ResNeXt50c32](tensornets/resnets.py#L184) | 22.260 | 6.190 | 5.410 | 25.1M | 23.0M | 267.4 | [[paper]](https://arxiv.org/abs/1611.05431) [[torch-fb]](https://github.com/facebookresearch/ResNeXt/blob/master/models/resnext.lua) |
| [ResNeXt101c32](tensornets/resnets.py#L200) | 21.270 | 5.706 | 4.842 | 44.3M | 42.3M | 427.9 | [[paper]](https://arxiv.org/abs/1611.05431) [[torch-fb]](https://github.com/facebookresearch/ResNeXt/blob/master/models/resnext.lua) |
Expand Down
3 changes: 2 additions & 1 deletion tensornets/resnets.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,8 @@ def resnet(x, preact, stack_fn, is_training, classes, stem,
else:
x = conv(x, 64, 7, stride=2, scope='conv1')
x = relu(x, name='conv1/relu')
x = pad(x, pad_info(0 if stem else 3), name='pool1/pad')
x = pad(x, pad_info(0 if stem else 3, symmetry=not preact),
'SYMMETRIC', name='pool1/pad')
x = max_pool2d(x, 3, stride=2, scope='pool1')
x = stack_fn(x)
if stem: return x
Expand Down

0 comments on commit c3b5320

Please sign in to comment.