louischenki/pytorch-se-resnet

SE-ResNet PyTorch Version

PyTorch-SE-ResNet

SE-ResNet PyTorch Version

I implement it for Place365 & Challenge AI Scene Classification.

Implemented Net

• SE-ResNet (50, 101, 152)
• SE-ResNeXt (50, 101, 152)

other framework cound be easily implement to modify model/model.py.

Source Caffe codes are at here

Citation

If you use Squeeze-and-Excitation Networks in your research, please cite the paper:

@article{hu2017,
  title={Squeeze-and-Excitation Networks},
  author={Jie Hu and Li Shen and Gang Sun},
  journal={arXiv preprint arXiv:1709.01507},
  year={2017}
}