Skip to content

SENet* models

Luigi edited this page Oct 1, 2018 · 2 revisions

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')
Clone this wiki locally