dakeli / senet.pytorch

PyTorch implementation of SENet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SENet.pytorch

An implementation of SENet, proposed in Squeeze-and-Excitation Networks by Jie Hu, Li Shen and Gang Sun, who are the winners of ILSVRC 2017 classification competition.

Now SE-ResNet (18, 34, 50, 101, 152) and SE-Inception-v3 are implemented.

python exec.py runs SE-ResNet50 with Cifar10 dataset.

For SE-Inception-v3, the input size is required to be 299x299 as original Inception.

references

paper authors' Caffe implementation

About

PyTorch implementation of SENet


Languages

Language:Python 100.0%