shicai / SENet-Caffe

A Caffe Re-Implementation of SENet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SENet-Caffe

Introduction

This is a Caffe implementation of Squeeze-and-Excitation Networks (SENet). For details, please read the original slides:

For offical implementations, please check this repo SENet.

Pretrained Models on ImageNet

Here we provide a pretrained SE-ResNet-50 model on ImageNet, which achieves slightly better accuracy rates than the original one reported in the official repo. You can use the official bvlc caffe to run this model without any modifications.

The top-1/5 accuracy rates by using single center crop (crop size: 224x224, image size: 256xN):

Network Top-1 Top-5 Download Architecture
SE-ResNet-50 78.01 94.21 caffemodel (107 MB) netscope, netron

For your convenience, we also provide a link to this model on Baidu Disk.

Notes

  • BGR mean values [103.94,116.78,123.68] are subtracted
  • scale: 0.017 is used as std values for image preprocessing
  • Images labels are the same as fb.resnet.torch. We also provide synset.txt, which can be found here.

About

A Caffe Re-Implementation of SENet

License:BSD 3-Clause "New" or "Revised" License