shicai / SENet-Caffe

A Caffe Re-Implementation of SENet
BSD 3-Clause "New" or "Revised" License
169 stars 101 forks source link
caffe imagenet resnet senet

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