Available pretrained models

May 15, 2018 ยท View on GitHub

Classification accuracy of available pretrained models. (Datasets are ImageNet1K, ImageNet11K and Place365 Challenge)

Please note that the following results are calculated with different datasets.

modelTop-1 AccuracyTop-5 Accuracydownload sizemodel size (MXNet)datasetimage shape
CaffeNet54.5%78.3%233MB9.3MBImageNet1K227x227
SqueezeNet55.4%78.8%4.8MB4.8MBImageNet1K227x227
NIN58.8%81.3%30MB30MBImageNet1K224x224
ResNet-1869.5%89.1%45MB43MBImageNet1K224x224
VGG1671.0%89.8%528MB58MBImageNet1K224x224
VGG1971.0%89.8%549MB78MBImageNet1K224x224
Inception-BN72.5%90.8%44MB40MBImageNet1K224x224
ResNet-3472.8%91.1%84MB82MBImageNet1K224x224
ResNet-5075.6%92.8%98MB90MBImageNet1K224x224
ResNet-10177.3%93.4%171MB163MBImageNet1K224x224
ResNet-15277.8%93.6%231MB223MBImageNet1K224x224
ResNet-20077.9%93.8%248MB240MBImageNet1K224x224
Inception-v376.9%93.3%92MB84MBImageNet1K299x299
ResNeXt-5076.9%93.3%96MB89MBImageNet1K224x224
ResNeXt-10178.3%94.1%170MB162MBImageNet1K224x224
ResNeXt-101-64x4d79.1%94.3%320MB312MBImageNet1K224x224
ResNet-152 (imagenet11k)41.6%-311MB223MBImageNet11K224x224
ResNet-50 (Place365 Challenge)31.1%-181MB90MBPlace365ch224x224
ResNet-152 (Place365 Challenge)33.6%-313MB223MBPlace365ch224x224
DenseNet-16975.3%92.8%55MB48MBImageNet1K224x224
SE-ResNeXt-5076.7%93.4%103MB95MBImageNet1K224x224
  • The download size is the file size when first downloading pretrained model.
  • The model size is the file size to be saved after fine-tuning.

To use these pretrained models, specify the following pretrained model name in config.yml.

modelpretrained model name
CaffeNetimagenet1k-caffenet
SqueezeNetimagenet1k-squeezenet
NINimagenet1k-nin
VGG16imagenet1k-vgg16
VGG19imagenet1k-vgg19
Inception-BNimagenet1k-inception-bn
ResNet-18imagenet1k-resnet-18
ResNet-34imagenet1k-resnet-34
ResNet-50imagenet1k-resnet-50
ResNet-101imagenet1k-resnet-101
ResNet-152imagenet1k-resnet-152
ResNet-152 (imagenet11k)imagenet11k-resnet-152
ResNet-200imagenet1k-resnet-200
Inception-v3imagenet1k-inception-v3
ResNeXt-50imagenet1k-resnext-50
ResNeXt-101imagenet1k-resnext-101
ResNeXt-101-64x4dimagenet1k-resnext-101-64x4d
ResNet-50 (Place365 Challenge)imagenet11k-place365ch-resnet-50
ResNet-152 (Place365 Challenge)imagenet11k-place365ch-resnet-152
DenseNet-169imagenet1k-densenet-169
SE-ResNeXt-50imagenet1k-se-resnext-50