SENet* models - CeLuigi/models-comparison.pytorch GitHub Wiki

J. Hu, L. Shen, and G. Sun, “Squeeze-and-excitation networks,” in Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2018.

@InProceedings{Hu_2018_CVPR,
author = {Hu, Jie and Shen, Li and Sun, Gang},
title = {Squeeze-and-Excitation Networks},
booktitle = {Conference on Computer Vision and Pattern Recognition (CVPR)},
year = {2018},
organization={IEEE}
}

Source: Caffe repo of Jie Hu

• senet154(num_classes=1000, pretrained='imagenet')
• se_resnet50(num_classes=1000, pretrained='imagenet')
• se_resnet101(num_classes=1000, pretrained='imagenet')
• se_resnet152(num_classes=1000, pretrained='imagenet')
• se_resnext50_32x4d(num_classes=1000, pretrained='imagenet')
• se_resnext101_32x4d(num_classes=1000, pretrained='imagenet')