Huge News!Announcing our $40M Series B led by Abstract Ventures.Learn More
Socket
Sign inDemoInstall
Socket

gluoncv2

Package Overview
Dependencies
Maintainers
1
Alerts
File Explorer

Advanced tools

Socket logo

Install Socket

Detect and block malicious and high-risk dependencies

Install

gluoncv2

Image classification and segmentation models for Gluon

  • 0.0.64
  • PyPI
  • Socket score

Maintainers
1

Computer vision models on MXNet/Gluon

PyPI Downloads

This is a collection of image classification, segmentation, detection, and pose estimation models. Many of them are pretrained on ImageNet-1K, CIFAR-10/100, SVHN, CUB-200-2011, Pascal VOC2012, ADE20K, Cityscapes, and COCO datasets and loaded automatically during use. All pretrained models require the same ordinary normalization. Scripts for training/evaluating/converting models are in the imgclsmob repo.

List of implemented models

Installation

To use the models in your project, simply install the gluoncv2 package with mxnet:

pip install gluoncv2 mxnet>=1.2.1

To enable different hardware supports such as GPUs, check out MXNet variants. For example, you can install with CUDA-9.2 supported MXNet:

pip install gluoncv2 mxnet-cu92>=1.2.1

Usage

Example of using a pretrained ResNet-18 model:

from gluoncv2.model_provider import get_model as glcv2_get_model
import mxnet as mx

net = glcv2_get_model("resnet18", pretrained=True)
x = mx.nd.zeros((1, 3, 224, 224), ctx=mx.cpu())
y = net(x)

Pretrained models

ImageNet-1K

Some remarks:

  • Top1/Top5 are the standard 1-crop Top-1/Top-5 errors (in percents) on the validation subset of the ImageNet-1K dataset.
  • FLOPs/2 is the number of FLOPs divided by two to be similar to the number of MACs.
  • ResNet/PreResNet with b-suffix is a version of the networks with the stride in the second convolution of the bottleneck block. Respectively a network without b-suffix has the stride in the first convolution.
  • ResNet/PreResNet models do not use biases in convolutions at all.
  • CondenseNet models are only so-called converted versions.
  • ShuffleNetV2 and ShuffleNetV2b are different implementations of the same architecture.
  • ResNet(A) is an average downsampled ResNet intended for use as an feature extractor in some pose estimation networks.
  • ResNet(D) is a dilated ResNet intended for use as an feature extractor in some segmentation networks.
  • Models with *-suffix use non-standard preprocessing (see the training log).
ModelTop1Top5ParamsFLOPs/2Remarks
AlexNet38.0716.1062,378,3441,132.33MTraining (log)
AlexNet-b39.3017.0561,100,840714.83MTraining (log)
ZFNet39.2116.7862,357,6081,170.33MTraining (log)
ZFNet-b35.8114.59107,627,6242,479.13MTraining (log)
VGG-1129.5910.16132,863,3367,615.87MTraining (log)
VGG-1328.379.50133,047,84811,317.65MTraining (log)
VGG-1626.618.32138,357,54415,480.10MTraining (log)
VGG-1925.587.67143,667,24019,642.55MTraining (log)
BN-VGG-1128.569.34132,866,0887,630.21MTraining (log)
BN-VGG-1327.688.87133,050,79211,341.62MTraining (log)
BN-VGG-1625.507.57138,361,76815,506.38MTraining (log)
BN-VGG-1923.916.89143,672,74419,671.15MTraining (log)
BN-VGG-11b29.249.75132,868,8407,630.72MTraining (log)
BN-VGG-13b28.239.12133,053,73611,342.14MTraining (log)
BN-VGG-16b25.837.75138,365,99215,507.20MTraining (log)
BN-VGG-19b24.797.35143,678,24819,672.26MTraining (log)
BN-Inception25.127.5411,295,2402,048.06MTraining (log)
ResNet-1032.5412.535,418,792894.04MTraining (log)
ResNet-1231.6812.035,492,7761,126.25MTraining (log)
ResNet-1430.3810.865,788,2001,357.94MTraining (log)
ResNet-BC-14b29.2210.3310,064,9361,479.12MTraining (log)
ResNet-1628.539.786,968,8721,589.34MTraining (log)
ResNet-18 x0.2539.3117.403,937,400270.94MTraining (log)
ResNet-18 x0.533.4112.845,804,296608.70MTraining (log)
ResNet-18 x0.7530.0010.668,476,0561,129.45MTraining (log)
ResNet-1826.798.6711,689,5121,820.41MTraining (log)
ResNet-2625.968.2317,960,2322,746.79MTraining (log)
ResNet-BC-26b24.867.5815,995,1762,356.67MTraining (log)
ResNet-3424.537.4321,797,6723,672.68MTraining (log)
ResNet-BC-38b23.506.7221,925,4163,234.21MTraining (log)
ResNet-5022.156.0425,557,0323,877.95MTraining (log)
ResNet-50b22.066.1125,557,0324,110.48MTraining (log)
ResNet-10120.525.1644,549,1607,597.95MTraining (log)
ResNet-101b20.265.1244,549,1607,830.48MTraining (log)
ResNet-15219.204.4460,192,80811,321.85MTraining (log)
ResNet-152b18.844.2960,192,80811,554.38MTraining (log)
PreResNet-1034.6514.015,417,128894.19MTraining (log)
PreResNet-1233.5713.215,491,1121,126.40MTraining (log)
PreResNet-1432.2912.185,786,5361,358.09MTraining (log)
PreResNet-BC-14b30.6711.5110,057,3841,476.62MTraining (log)
PreResNet-1630.2110.816,967,2081,589.49MTraining (log)
PreResNet-18 x0.2539.6217.783,935,960270.93MTraining (log)
PreResNet-18 x0.533.6713.195,802,440608.73MTraining (log)
PreResNet-18 x0.7529.9610.688,473,7841,129.51MTraining (log)
PreResNet-1828.169.5111,687,8481,820.56MTraining (log)
PreResNet-2626.038.3417,958,5682,746.94MTraining (log)
PreResNet-BC-26b25.217.8615,987,6242,354.16MTraining (log)
PreResNet-3424.557.5121,796,0083,672.83MTraining (log)
PreResNet-BC-38b22.676.3321,917,8643,231.70MTraining (log)
PreResNet-5022.276.2025,549,4803,875.44MTraining (log)
PreResNet-50b22.366.3225,549,4804,107.97MTraining (log)
PreResNet-10120.605.3444,541,6087,595.44MTraining (log)
PreResNet-101b20.855.4044,541,6087,827.97MTraining (log)
PreResNet-15219.174.4660,185,25611,319.34MTraining (log)
PreResNet-152b19.014.3860,185,25611,551.87MTraining (log)
PreResNet-200b18.964.4664,666,28015,068.63MTraining (log)
PreResNet-269b20.175.01102,065,83220,101.11MTraining (log)
ResNeXt-14 (16x4d)31.6612.237,127,3361,045.77MTraining (log)
ResNeXt-14 (32x2d)32.1612.477,029,4161,031.32MTraining (log)
ResNeXt-14 (32x4d)29.9511.109,411,8801,603.46MTraining (log)
ResNeXt-26 (32x2d)26.348.509,924,1361,461.06MTraining (log)
ResNeXt-26 (32x4d)23.937.2115,389,4802,488.07MTraining (log)
ResNeXt-50 (32x4d)20.845.4525,028,9044,255.86MTraining (log)
ResNeXt-101 (32x4d)18.464.1844,177,7048,003.45MTraining (log)
ResNeXt-101 (64x4d)18.804.3983,455,27215,500.27MTraining (log)
SE-ResNet-1031.3611.695,463,332894.08MTraining (log)
SE-ResNet-1231.6411.765,537,8961,126.29MTraining (log)
SE-ResNet-1430.3410.955,835,5041,357.99MTraining (log)
SE-ResNet-1628.419.727,024,6401,589.40MTraining (log)
SE-ResNet-1827.959.2011,778,5921,820.51MTraining (log)
SE-ResNet-2625.428.0318,093,8522,746.93MTraining (log)
SE-ResNet-BC-26b23.446.8217,395,9762,358.07MTraining (log)
SE-ResNet-BC-38b21.445.7524,026,6163,236.32MTraining (log)
SE-ResNet-5021.075.6028,088,0243,880.49MTraining (log)
SE-ResNet-50b20.585.3328,088,0244,113.02MTraining (log)
SE-ResNet-10119.004.4149,326,8727,602.76MTraining (log)
SE-ResNet-101b19.464.6249,326,8727,835.29MTraining (log)
SE-ResNet-15218.594.3066,821,84811,328.52MTraining (log)
SE-PreResNet-1032.3712.215,461,668894.23MTraining (log)
SE-PreResNet-1231.5811.805,536,2321,126.44MTraining (log)
SE-PreResNet-1628.399.567,022,9761,589.55MTraining (log)
SE-PreResNet-1827.168.8111,776,9281,820.66MTraining (log)
SE-PreResNet-2625.958.0418,092,1882,747.08MTraining (log)
SE-PreResNet-BC-26b22.956.3617,388,4242,355.57MTraining (log)
SE-PreResNet-BC-38b21.425.6324,019,0643,233.81MTraining (log)
SE-PreResNet-50b20.675.3228,080,4724,110.51MTraining (log)
SE-ResNeXt-50 (32x4d)18.744.3327,559,8964,258.40MTraining (log)
SE-ResNeXt-101 (32x4d)19.064.4448,955,4168,008.26MTraining (log)
SE-ResNeXt-101 (64x4d)18.434.0888,232,98415,505.08MTraining (log)
SENet-1625.348.0631,366,1685,080.55MTraining (log)
SENet-2821.685.9136,453,7685,731.20MTraining (log)
SENet-15418.844.40115,088,98420,745.78MTraining (log)
ResNeSt(A)-BC-1422.276.3410,611,6882,766.86MTraining (log)
ResNeSt(A)-1823.436.8912,763,7842,587.11MTraining (log)
ResNeSt(A)-BC-2619.574.7017,069,4483,645.87MTraining (log)
ResNeSt(A)-5018.924.3827,483,2405,402.09MTraining (log)
ResNeSt(A)-10117.743.9948,275,01610,246.42MFrom dmlc/gluon-cv (log)
ResNeSt(A)-15218.724.5165,316,04013,974.33MTraining (log)
ResNeSt(A)-20016.783.4070,201,54422,854.22MFrom dmlc/gluon-cv (log)
ResNeSt(A)-26916.383.36110,929,48046,005.88MFrom dmlc/gluon-cv (log)
IBN-ResNet-5021.465.5925,557,0324,110.48MTraining (log)
IBN-ResNet-10119.694.8944,549,1607,830.48MTraining (log)
IBN(b)-ResNet-5021.705.7925,558,5684,112.89MTraining (log)
IBN-ResNeXt-101 (32x4d)19.764.8744,177,7048,003.45MTraining (log)
IBN-DenseNet-12123.336.467,978,8562,872.13MTraining (log)
IBN-DenseNet-16922.146.0814,149,4803,403.89MTraining (log)
AirNet50-1x64d (r=2)20.455.2327,425,8644,772.11MTraining (log)
AirNet50-1x64d (r=16)21.115.4425,714,9524,399.97MTraining (log)
AirNeXt50-32x4d (r=2)19.845.0427,604,2965,339.58MTraining (log)
BAM-ResNet-5020.595.3825,915,0994,196.09MTraining (log)
CBAM-ResNet-5019.944.8828,089,6244,116.97MTraining (log)
SCNet-5020.535.1125,564,5843,951.01MTraining (log)
SCNet-10119.214.4644,565,4167,204.19MTraining (log)
SCNet(A)-5019.014.5925,583,8164,715.79MTraining (log)
RegNetX-200MF29.9110.382,684,792203.32MTraining (log)
RegNetX-400MF26.258.555,157,512403.44MTraining (log)
RegNetX-600MF24.717.566,196,040608.36MTraining (log)
RegNetX-800MF24.097.247,259,656809.47MTraining (log)
RegNetX-1.6GF22.126.139,190,1361,618.97MTraining (log)
RegNetX-3.2GF21.285.6815,296,5523,199.52MTraining (log)
RegNetX-4.0GF19.514.6922,118,2483,986.26MTraining (log)
RegNetX-6.4GF19.224.5826,209,2566,490.97MTraining (log)
RegNetX-8.0GF19.624.6639,572,6488,017.90MTraining (log)
RegNetX-12GF19.995.1846,106,05612,124.16MTraining (log)
RegNetX-16GF19.124.5654,278,53615,986.59MTraining (log)
RegNetX-32GF17.833.94107,811,56031,790.18MTraining (log)
RegNetY-200MF28.509.533,162,996203.80MTraining (log)
RegNetY-400MF24.857.474,344,144409.95MTraining (log)
RegNetY-600MF23.596.976,055,160609.91MTraining (log)
RegNetY-800MF22.546.456,263,168808.07MTraining (log)
RegNetY-1.6GF21.235.6811,202,4301,628.43MTraining (log)
RegNetY-3.2GF18.324.1319,436,3383,197.70MFrom rwightman/pyt...models (log)
RegNetY-4.0GF19.564.6720,646,6563,997.63MTraining (log)
RegNetY-6.4GF18.954.4530,583,2526,386.79MTraining (log)
RegNetY-8.0GF18.784.3639,180,0687,994.33MTraining (log)
RegNetY-12GF18.514.3151,822,54412,129.89MTraining (log)
RegNetY-16GF18.634.3083,590,14015,941.65MTraining (log)
RegNetY-32GF17.833.73145,046,77032,313.76MTraining (log)
PyramidNet-101 (a=360)20.415.2042,455,0708,743.54MTraining (log)
DiracNetV2-1830.6111.1711,511,7841,796.62MFrom szagoruyko/diracnets (log)
DiracNetV2-3427.939.4621,616,2323,646.93MFrom szagoruyko/diracnets (log)
CRU-Net-5620.645.3625,609,3845,660.66MTraining (log)
DenseNet-12123.256.857,978,8562,872.13MTraining (log)
DenseNet-16121.825.9228,681,0007,793.16MTraining (log)
DenseNet-16922.106.0514,149,4803,403.89MTraining (log)
DenseNet-20121.565.9020,013,9284,347.15MTraining (log)
CondenseNet-74 (C=G=4)26.828.644,773,944546.06MFrom ShichenLiu/CondenseNet (log)
CondenseNet-74 (C=G=8)29.7610.492,935,416291.52MFrom ShichenLiu/CondenseNet (log)
PeleeNet29.389.792,802,248514.87MTraining (log)
WRN-50-222.026.0668,849,12811,405.42MTraining (log)
DRN-C-2624.327.1121,126,58416,993.90MTraining (log)
DRN-C-4222.256.1431,234,74425,093.75MTraining (log)
DRN-C-5820.475.1540,542,00832,489.94MTraining (log)
DRN-D-2224.677.4416,393,75213,051.33MTraining (log)
DRN-D-3822.836.2426,501,91221,151.19MTraining (log)
DRN-D-5420.294.9735,809,17628,547.38MTraining (log)
DRN-D-10521.315.8154,801,30443,442.43MFrom fyu/drn (log)
DPN-6822.876.5812,611,6022,351.84MTraining (log)
DPN-9818.294.2261,570,72811,716.51MTraining (log)
DPN-13119.424.7779,254,50416,076.15MTraining (log)
DarkNet Tiny40.3117.461,042,104500.85MTraining (log)
DarkNet Ref38.0016.687,319,416367.59MTraining (log)
DarkNet-5321.275.5041,609,9287,133.86MTraining (log)
i-RevNet-30126.978.97125,120,35614,453.87MFrom jhjacobsen/pytorch-i-revnet (log)
BagNet-948.7725.3615,688,74416,049.19MTraining (log)
BagNet-1736.5115.2316,213,03215,768.77MTraining (log)
BagNet-3329.4910.4118,310,18416,371.52MTraining (log)
DLA-3424.317.0515,742,1043,071.37MTraining (log)
DLA-46-C33.8412.861,301,400585.45MTraining (log)
DLA-X-46-C32.9612.251,068,440546.72MTraining (log)
DLA-6021.275.5422,036,6324,255.49MTraining (log)
DLA-X-6020.705.5317,352,3443,543.68MTraining (log)
DLA-X-60-C30.6710.741,319,832596.06MTraining (log)
DLA-10220.585.1733,268,8887,190.95MTraining (log)
DLA-X-10219.594.7026,309,2725,884.94MTraining (log)
DLA-X2-10218.664.2341,282,2009,340.61MTraining (log)
DLA-16919.284.6053,389,72011,593.20MTraining (log)
FishNet-15019.154.6624,959,4006,435.02MTraining (log)
ESPNetv2 x0.543.3820.821,241,33235.36MTraining (log)
ESPNetv2 x1.035.3314.271,670,07298.09MFrom sacmehta/ESPNetv2 (log)
ESPNetv2 x1.2533.1412.731,965,440138.18MFrom sacmehta/ESPNetv2 (log)
ESPNetv2 x1.531.0511.352,314,856185.77MTraining (log)
ESPNetv2 x2.028.919.943,498,136306.93MFrom sacmehta/ESPNetv2 (log)
DiCENet x0.255.1530.671,130,70418.70MFrom sacmehta/EdgeNets (log)
DiCENet x0.547.1523.081,214,12030.39MTraining (log)
DiCENet x0.7538.2516.471,495,67655.64MFrom sacmehta/EdgeNets (log)
DiCENet x1.035.0214.111,805,60481.96MTraining (log)
DiCENet x1.2533.1112.512,162,888111.60MTraining (log)
DiCENet x1.531.0011.442,652,200151.48MTraining (log)
DiCENet x1.7530.0810.813,264,932200.87MTraining (log)
DiCENet x2.029.9310.583,979,044257.49MFrom sacmehta/EdgeNets (log)
HRNet-W18 Small V126.208.7313,187,4641,614.87MTraining (log)
HRNet-W18 Small V221.716.0215,597,4642,618.54MTraining (log)
HRNetV2-W1820.155.0021,299,0044,322.66MTraining (log)
HRNetV2-W3020.305.0837,712,2208,156.14MTraining (log)
HRNetV2-W3219.944.9641,232,6808,973.31MTraining (log)
HRNetV2-W4019.654.8157,557,16012,751.34MTraining (log)
HRNetV2-W4419.674.8667,064,98414,945.95MTraining (log)
HRNetV2-W4819.464.8477,469,86417,344.29MTraining (log)
HRNetV2-W6419.504.78128,059,94428,974.95MTraining (log)
VoVNet-27-slim29.289.803,525,7362,187.25MTraining (log)
VoVNet-3921.545.4822,600,2967,086.16MTraining (log)
VoVNet-5720.145.1036,640,2968,943.09MTraining (log)
SelecSLS-42b21.725.9632,458,2482,980.62MTraining (log)
SelecSLS-6020.205.1130,670,7683,591.78MTraining (log)
SelecSLS-60b20.625.3732,774,0643,629.14MTraining (log)
HarDNet-39DS26.528.643,488,228437.52MTraining (log)
HarDNet-68DS24.257.384,180,602788.86MTraining (log)
HarDNet-6824.037.0217,565,3484,256.32MTraining (log)
HarDNet-8521.875.7236,670,2129,088.58MTraining (log)
SqueezeNet v1.038.7317.341,248,424823.67MTraining (log)
SqueezeNet v1.139.0917.391,235,496352.02MTraining (log)
SqueezeResNet v1.039.3217.671,248,424823.67MTraining (log)
SqueezeResNet v1.139.8317.841,235,496352.02MTraining (log)
1.0-SqNxt-2342.2518.66724,056287.28MTraining (log)
1.0-SqNxt-23v540.4317.43921,816285.82MTraining (log)
1.5-SqNxt-2334.4613.211,511,824552.39MTraining (log)
1.5-SqNxt-23v533.4812.681,953,616550.97MTraining (log)
2.0-SqNxt-2330.2410.632,583,752898.48MTraining (log)
2.0-SqNxt-23v529.2710.243,366,344897.60MTraining (log)
ShuffleNet x0.25 (g=1)62.0036.77209,74612.35MTraining (log)
ShuffleNet x0.25 (g=3)61.3436.17305,90213.09MTraining (log)
ShuffleNet x0.5 (g=1)46.2222.38534,48441.16MTraining (log)
ShuffleNet x0.5 (g=3)43.8320.60718,32441.70MTraining (log)
ShuffleNet x0.75 (g=1)39.2516.75975,21486.42MTraining (log)
ShuffleNet x0.75 (g=3)37.8116.091,238,26685.82MTraining (log)
ShuffleNet x1.0 (g=1)34.4113.501,531,936148.13MTraining (log)
ShuffleNet x1.0 (g=2)33.9813.321,733,848147.60MTraining (log)
ShuffleNet x1.0 (g=3)33.9613.291,865,728145.46MTraining (log)
ShuffleNet x1.0 (g=4)33.8413.101,968,344143.33MTraining (log)
ShuffleNet x1.0 (g=8)33.6513.192,434,768150.76MTraining (log)
ShuffleNetV2 x0.540.6118.301,366,79243.31MTraining (log)
ShuffleNetV2 x1.030.9411.232,278,604149.72MTraining (log)
ShuffleNetV2 x1.527.179.134,406,098320.77MTraining (log)
ShuffleNetV2 x2.025.808.237,601,686595.84MTraining (log)
ShuffleNetV2b x0.539.8117.821,366,79243.31MTraining (log)
ShuffleNetV2b x1.030.3911.012,279,760150.62MTraining (log)
ShuffleNetV2b x1.526.908.794,410,194323.98MTraining (log)
ShuffleNetV2b x2.025.208.107,611,290603.37MTraining (log)
108-MENet-8x1 (g=3)43.6220.30654,51642.68MTraining (log)
128-MENet-8x1 (g=4)42.1019.13750,79645.98MTraining (log)
128-MENet-8x1 (g=4)42.1019.13750,79645.98MTraining (log)
160-MENet-8x1 (g=8)43.4720.28850,12045.63MTraining (log)
256-MENet-12x1 (g=4)32.2312.161,888,240150.65MTraining (log)
348-MENet-12x1 (g=3)27.859.363,368,128312.00MTraining (log)
352-MENet-12x1 (g=8)31.3011.672,272,872157.35MTraining (log)
456-MENet-24x1 (g=3)25.027.805,304,784567.90MTraining (log)
MobileNet x0.2545.7822.18470,07244.09MTraining (log)
MobileNet x0.533.9413.301,331,592155.42MTraining (log)
MobileNet x0.7529.8510.512,585,560333.99MTraining (log)
MobileNet x1.026.438.654,231,976579.80MTraining (log)
MobileNetb x0.2545.2521.65467,59242.88MTraining (log)
MobileNetb x0.532.8912.711,326,632153.00MTraining (log)
MobileNetb x0.7529.0810.202,578,120330.37MTraining (log)
MobileNetb x1.025.067.884,222,056574.97MTraining (log)
FD-MobileNet x0.2555.4430.53383,16012.95MTraining (log)
FD-MobileNet x0.542.6219.69993,92841.84MTraining (log)
FD-MobileNet x0.7537.9116.011,833,30486.68MTraining (log)
FD-MobileNet x1.033.8013.122,901,288147.46MTraining (log)
MobileNetV2 x0.2548.0824.121,516,39234.24MTraining (log)
MobileNetV2 x0.535.6314.421,964,736100.13MTraining (log)
MobileNetV2 x0.7529.7810.442,627,592198.50MTraining (log)
MobileNetV2 x1.026.778.643,504,960329.36MTraining (log)
MobileNetV2b x0.2546.7223.381,516,31233.18MTraining (log)
MobileNetV2b x0.534.2613.731,964,44896.42MTraining (log)
MobileNetV2b x0.7530.1910.642,626,968190.52MTraining (log)
MobileNetV2b x1.027.168.843,503,872315.51MTraining (log)
MobileNetV3 L/224/1.024.367.295,481,752226.80MTraining (log)
IGCV3 x0.2553.4328.301,534,02041.29MTraining (log)
IGCV3 x0.539.4117.031,985,528111.12MTraining (log)
IGCV3 x0.7530.7110.962,638,084210.95MTraining (log)
IGCV3 x1.027.739.003,491,688340.79MTraining (log)
MnasNet-B124.677.234,383,312326.30MTraining (log)
MnasNet-A124.047.053,887,038325.77MTraining (log)
DARTS24.917.564,718,752539.86MTraining (log)
ProxylessNAS CPU24.787.504,361,648459.96MTraining (log)
ProxylessNAS GPU24.677.247,119,848476.08MTraining (log)
ProxylessNAS Mobile25.317.804,080,512332.46MTraining (log)
ProxylessNAS Mob-1422.966.516,857,568597.10MTraining (log)
FBNet-Cb24.867.615,572,200399.26MTraining (log)
Xception20.435.1122,855,9528,403.63MTraining (log)
InceptionV320.515.3323,834,5685,743.06MTraining (log)
InceptionV419.834.8842,679,81612,304.93MTraining (log)
InceptionResNetV119.564.8123,995,6246,329.60MTraining (log)
InceptionResNetV219.484.7055,843,46413,188.64MTraining (log)
PolyNet19.094.5395,366,60034,821.34MFrom Cadene/pretrained...pytorch (log)
NASNet-A 4@105625.167.905,289,978584.90MTraining (log)
NASNet-A 6@403218.174.2488,753,15023,976.44MFrom Cadene/pretrained...pytorch (log)
PNASNet-5-Large17.904.2886,057,66825,140.77MFrom Cadene/pretrained...pytorch (log)
SPNASNet25.107.764,421,616346.73MTraining (log)
EfficientNet-B024.507.225,288,548413.13MTraining (log)
EfficientNet-B122.896.267,794,184730.44MTraining (log)
EfficientNet-B0b22.966.705,288,548413.13MFrom rwightman/pyt...models (log)
EfficientNet-B1b20.985.657,794,184730.44MFrom rwightman/pyt...models (log)
EfficientNet-B2b19.945.169,109,9941,049.29MFrom rwightman/pyt...models (log)
EfficientNet-B3b18.604.3112,233,2321,923.98MFrom rwightman/pyt...models (log)
EfficientNet-B4b17.253.7619,341,6164,597.56MFrom rwightman/pyt...models (log)
EfficientNet-B5b16.393.3430,389,78410,674.67MFrom rwightman/pyt...models (log)
EfficientNet-B6b15.963.1243,040,70419,761.35MFrom rwightman/pyt...models (log)
EfficientNet-B7b15.703.1166,347,96038,949.07MFrom rwightman/pyt...models (log)
EfficientNet-B0c*22.526.465,288,548413.13MFrom rwightman/pyt...models (log)
EfficientNet-B1c*20.505.557,794,184730.44MFrom rwightman/pyt...models (log)
EfficientNet-B2c*19.604.899,109,9941,049.29MFrom rwightman/pyt...models (log)
EfficientNet-B3c*18.194.3412,233,2321,923.98MFrom rwightman/pyt...models (log)
EfficientNet-B4c*16.743.5919,341,6164,597.56MFrom rwightman/pyt...models (log)
EfficientNet-B5c*15.793.0230,389,78410,674.67MFrom rwightman/pyt...models (log)
EfficientNet-B6c*15.292.8543,040,70419,761.35MFrom rwightman/pyt...models (log)
EfficientNet-B7c*14.872.7766,347,96038,949.07MFrom rwightman/pyt...models (log)
EfficientNet-B8c*14.612.7087,413,14264,446.06MFrom rwightman/pyt...models (log)
EfficientNet-Edge-Small-b*22.486.295,438,3922,378.09MFrom rwightman/pyt...models (log)
EfficientNet-Edge-Medium-b*21.085.536,899,4963,700.08MFrom rwightman/pyt...models (log)
EfficientNet-Edge-Large-b*19.504.7710,589,7129,747.58MFrom rwightman/pyt...models (log)
MixNet-S23.837.034,134,606260.26MTraining (log)
MixNet-M22.376.315,014,382366.05MTraining (log)
MixNet-L21.485.577,329,252590.45MTraining (log)
ResNet(A)-1030.8911.595,438,0241,135.85MTraining (log)
ResNet(A)-BC-1427.759.5610,084,1681,721.52MTraining (log)
ResNet(A)-1825.388.0211,708,7442,062.22MTraining (log)
ResNet(A)-50b20.785.3425,576,2644,352.88MTraining (log)
ResNet(A)-101b18.984.4244,568,3928,072.88MTraining (log)
ResNet(A)-152b18.584.2460,212,04011,796.78MTraining (log)
ResNet(D)-50b20.795.4925,680,80820,496.80MFrom dmlc/gluon-cv (log)
ResNet(D)-101b19.494.6144,672,93635,391.85MFrom dmlc/gluon-cv (log)
ResNet(D)-152b19.394.6760,316,58447,661.38MFrom dmlc/gluon-cv (log)

CIFAR-10

Some remarks:

  • Testing subset is used for validation purpose.
  • Features means feature extractor output size.
ModelError, %FeaturesParamsFLOPs/2Remarks
NIN7.43192966,986222.97MTraining (log)
ResNet-205.9764272,47441.29MTraining (log)
ResNet-564.5264855,770127.06MTraining (log)
ResNet-1103.69641,730,714255.70MTraining (log)
ResNet-164(BN)3.682561,704,154255.31MTraining (log)
ResNet-272(BN)3.332562,816,986420.61MTraining (log)
ResNet-542(BN)3.432565,599,066833.87MTraining (log)
ResNet-10013.2825610,328,6021,536.40MTraining (log)
ResNet-12023.536419,424,0262,857.17MTraining (log)
PreResNet-206.5164272,28241.27MTraining (log)
PreResNet-564.4964855,578127.03MTraining (log)
PreResNet-1103.86641,730,522255.68MTraining (log)
PreResNet-164(BN)3.642561,703,258255.08MTraining (log)
PreResNet-272(BN)3.252562,816,090420.38MTraining (log)
PreResNet-542(BN)3.142565,598,170833.64MTraining (log)
PreResNet-10012.6525610,327,7061,536.18MTraining (log)
PreResNet-12023.396419,423,8342,857.14MTraining (log)
ResNeXt-20 (1x64d)4.3310243,446,602538.36MTraining (log)
ResNeXt-20 (2x32d)4.5310242,672,458425.15MTraining (log)
ResNeXt-20 (4x16d)4.7010242,285,386368.55MTraining (log)
ResNeXt-20 (8x8d)4.6610242,091,850340.25MTraining (log)
ResNeXt-20 (16x4d)4.0410241,995,082326.10MTraining (log)
ResNeXt-20 (32x2d)4.6110241,946,698319.03MTraining (log)
ResNeXt-20 (64x1d)4.9310241,922,506315.49MTraining (log)
ResNeXt-20 (2x64d)4.0310246,198,602987.98MTraining (log)
ResNeXt-20 (4x32d)3.7310244,650,314761.57MTraining (log)
ResNeXt-20 (8x16d)4.0410243,876,170648.37MTraining (log)
ResNeXt-20 (16x8d)3.9410243,489,098591.77MTraining (log)
ResNeXt-20 (32x4d)4.2010243,295,562563.47MTraining (log)
ResNeXt-20 (64x2d)4.3810243,198,794549.32MTraining (log)
ResNeXt-56 (1x64d)2.8710249,317,1941,399.33MTraining (log)
ResNeXt-56 (2x32d)3.0110246,994,7621,059.72MTraining (log)
ResNeXt-56 (4x16d)3.1110245,833,546889.91MTraining (log)
ResNeXt-56 (8x8d)3.0710245,252,938805.01MTraining (log)
ResNeXt-56 (16x4d)3.1210244,962,634762.56MTraining (log)
ResNeXt-56 (32x2d)3.1410244,817,482741.34MTraining (log)
ResNeXt-56 (64x1d)3.4110244,744,906730.72MTraining (log)
ResNeXt-29 (32x4d)3.1510244,775,754780.55MTraining (log)
ResNeXt-29 (16x64d)2.41102468,155,21010,709.34MTraining (log)
ResNeXt-272 (1x64d)2.55102444,540,7466,565.15MTraining (log)
ResNeXt-272 (2x32d)2.74102432,928,5864,867.11MTraining (log)
SE-ResNet-206.0164274,84741.30MTraining (log)
SE-ResNet-564.1364862,889127.07MTraining (log)
SE-ResNet-1103.63641,744,952255.72MTraining (log)
SE-ResNet-164(BN)3.392561,906,258255.52MTraining (log)
SE-ResNet-272(BN)3.392563,153,826420.96MTraining (log)
SE-ResNet-542(BN)3.472566,272,746834.57MTraining (log)
SE-PreResNet-206.1864274,55941.30MTraining (log)
SE-PreResNet-564.5164862,601127.07MTraining (log)
SE-PreResNet-1104.54641,744,664255.72MTraining (log)
SE-PreResNet-164(BN)3.732561,904,882255.29MTraining (log)
SE-PreResNet-272(BN)3.392563,152,450420.73MTraining (log)
SE-PreResNet-542(BN)3.082566,271,370834.34MTraining (log)
PyramidNet-110 (a=48)3.72641,772,706408.37MTraining (log)
PyramidNet-110 (a=84)2.981003,904,446778.15MTraining (log)
PyramidNet-110 (a=270)2.5128628,485,4774,730.60MTraining (log)
PyramidNet-164 (a=270, BN)2.42114427,216,0214,608.81MTraining (log)
PyramidNet-200 (a=240, BN)2.44102426,752,7024,563.40MTraining (log)
PyramidNet-236 (a=220, BN)2.4794426,969,0464,631.32MTraining (log)
PyramidNet-272 (a=200, BN)2.3986426,210,8424,541.36MTraining (log)
DenseNet-40 (k=12)5.61258599,050210.80MTraining (log)
DenseNet-BC-40 (k=12)6.43132176,12274.89MTraining (log)
DenseNet-BC-40 (k=24)4.52264690,346293.09MTraining (log)
DenseNet-BC-40 (k=36)4.043961,542,682654.60MTraining (log)
DenseNet-100 (k=12)3.666784,068,4901,353.55MTraining (log)
DenseNet-100 (k=24)3.13135616,114,1385,354.19MTraining (log)
DenseNet-BC-100 (k=12)4.16342769,162298.45MTraining (log)
DenseNet-BC-190 (k=40)2.52219025,624,4309,400.45MTraining (log)
DenseNet-BC-250 (k=24)2.67173415,324,4065,519.54MTraining (log)
X-DenseNet-BC-40-2 (k=24)5.31264690,346293.09MTraining (log)
X-DenseNet-BC-40-2 (k=36)4.373961,542,682654.60MTraining (log)
WRN-16-102.9364017,116,6342,414.04MTraining (log)
WRN-28-102.3964036,479,1945,246.98MTraining (log)
WRN-40-82.3751235,748,3145,176.90MTraining (log)
WRN-20-10-1bit3.2664026,737,1404,019.14MTraining (log)
WRN-20-10-32bit3.1464026,737,1404,019.14MTraining (log)
RoR-3-565.4364762,746113.43MTraining (log)
RoR-3-1104.35641,637,690242.07MTraining (log)
RoR-3-1643.93642,512,634370.72MTraining (log)
RiR3.283849,492,9801,281.08MTraining (log)
Shake-Shake-ResNet-20-2x16d5.1564541,08281.78MTraining (log)
Shake-Shake-ResNet-26-2x32d3.17642,923,162428.89MTraining (log)
DIA-ResNet-206.2264286,86641.34MTraining (log)
DIA-ResNet-565.0564870,162127.18MTraining (log)
DIA-ResNet-1104.10641,745,106255.94MTraining (log)
DIA-ResNet-164(BN)3.502561,923,002259.18MTraining (log)
DIA-PreResNet-206.4264286,67441.31MTraining (log)
DIA-PreResNet-564.8364869,970127.15MTraining (log)
DIA-PreResNet-1104.25641,744,914255.92MTraining (log)
DIA-PreResNet-164(BN)3.562561,922,106258.95MTraining (log)

CIFAR-100

Some remarks:

  • Testing subset is used for validation purpose.
ModelError, %ParamsFLOPs/2Remarks
NIN28.39984,356224.08MTraining (log)
ResNet-2029.64278,32441.30MTraining (log)
ResNet-5624.88861,620127.06MTraining (log)
ResNet-11022.801,736,564255.71MTraining (log)
ResNet-164(BN)20.441,727,284255.33MTraining (log)
ResNet-272(BN)20.072,840,116420.63MTraining (log)
ResNet-542(BN)19.325,622,196833.89MTraining (log)
ResNet-100119.7910,351,7321,536.43MTraining (log)
ResNet-120221.5619,429,8762,857.17MTraining (log)
PreResNet-2030.22278,13241.28MTraining (log)
PreResNet-5625.05861,428127.04MTraining (log)
PreResNet-11022.671,736,372255.68MTraining (log)
PreResNet-164(BN)20.181,726,388255.10MTraining (log)
PreResNet-272(BN)19.632,839,220420.40MTraining (log)
PreResNet-542(BN)18.715,621,300833.66MTraining (log)
PreResNet-100118.4110,350,8361,536.20MTraining (log)
ResNeXt-20 (1x64d)21.973,538,852538.45MTraining (log)
ResNeXt-20 (2x32d)22.552,764,708425.25MTraining (log)
ResNeXt-20 (4x16d)23.042,377,636368.65MTraining (log)
ResNeXt-20 (8x8d)22.822,184,100340.34MTraining (log)
ResNeXt-20 (16x4d)22.822,087,332326.19MTraining (log)
ResNeXt-20 (32x2d)21.732,038,948319.12MTraining (log)
ResNeXt-20 (64x1d)23.532,014,756315.58MTraining (log)
ResNeXt-20 (2x64d)20.606,290,852988.07MTraining (log)
ResNeXt-20 (4x32d)21.314,742,564761.66MTraining (log)
ResNeXt-20 (8x16d)21.723,968,420648.46MTraining (log)
ResNeXt-20 (16x8d)21.733,581,348591.86MTraining (log)
ResNeXt-20 (32x4d)22.133,387,812563.56MTraining (log)
ResNeXt-20 (64x2d)22.353,291,044549.41MTraining (log)
ResNeXt-56 (1x64d)18.259,409,4441,399.42MTraining (log)
ResNeXt-56 (2x32d)17.867,087,0121,059.81MTraining (log)
ResNeXt-56 (4x16d)18.095,925,796890.01MTraining (log)
ResNeXt-56 (8x8d)18.065,345,188805.10MTraining (log)
ResNeXt-56 (16x4d)18.245,054,884762.65MTraining (log)
ResNeXt-56 (32x2d)18.604,909,732741.43MTraining (log)
ResNeXt-56 (64x1d)18.164,837,156730.81MTraining (log)
ResNeXt-29 (32x4d)19.504,868,004780.64MTraining (log)
ResNeXt-29 (16x64d)16.9368,247,46010,709.43MTraining (log)
ResNeXt-272 (1x64d)19.1144,632,9966,565.25MTraining (log)
ResNeXt-272 (2x32d)18.3433,020,8364,867.20MTraining (log)
SE-ResNet-2028.54280,69741.30MTraining (log)
SE-ResNet-5622.94868,739127.07MTraining (log)
SE-ResNet-11020.861,750,802255.72MTraining (log)
SE-ResNet-164(BN)19.951,929,388255.54MTraining (log)
SE-ResNet-272(BN)19.073,176,956420.98MTraining (log)
SE-ResNet-542(BN)18.876,295,876834.59MTraining (log)
SE-PreResNet-2028.31280,40941.31MTraining (log)
SE-PreResNet-5623.05868,451127.08MTraining (log)
SE-PreResNet-11022.611,750,514255.73MTraining (log)
SE-PreResNet-164(BN)20.051,928,012255.31MTraining (log)
SE-PreResNet-272(BN)19.133,175,580420.75MTraining (log)
SE-PreResNet-542(BN)19.456,294,500834.36MTraining (log)
PyramidNet-110 (a=48)20.951,778,556408.38MTraining (log)
PyramidNet-110 (a=84)18.873,913,536778.16MTraining (log)
PyramidNet-110 (a=270)17.1028,511,3074,730.62MTraining (log)
PyramidNet-164 (a=270, BN)16.7027,319,0714,608.91MTraining (log)
PyramidNet-200 (a=240, BN)16.0926,844,9524,563.49MTraining (log)
PyramidNet-236 (a=220, BN)16.3427,054,0964,631.41MTraining (log)
PyramidNet-272 (a=200, BN)16.1926,288,6924,541.43MTraining (log)
DenseNet-40 (k=12)24.90622,360210.82MTraining (log)
DenseNet-BC-40 (k=12)28.41188,09274.90MTraining (log)
DenseNet-BC-40 (k=24)22.67714,196293.11MTraining (log)
DenseNet-BC-40 (k=36)20.501,578,412654.64MTraining (log)
DenseNet-100 (k=12)19.644,129,6001,353.62MTraining (log)
DenseNet-100 (k=24)18.0816,236,2685,354.32MTraining (log)
DenseNet-BC-100 (k=12)21.19800,032298.48MTraining (log)
DenseNet-BC-250 (k=24)17.3915,480,5565,519.69MTraining (log)
X-DenseNet-BC-40-2 (k=24)23.96714,196293.11MTraining (log)
X-DenseNet-BC-40-2 (k=36)21.651,578,412654.64MTraining (log)
WRN-16-1018.9517,174,3242,414.09MTraining (log)
WRN-28-1017.8836,536,8845,247.04MTraining (log)
WRN-40-818.0335,794,4845,176.95MTraining (log)
WRN-20-10-1bit19.0426,794,9204,022.81MTraining (log)
WRN-20-10-32bit18.1226,794,9204,022.81MTraining (log)
RoR-3-5625.49768,596113.43MTraining (log)
RoR-3-11023.641,643,540242.08MTraining (log)
RoR-3-16422.342,518,484370.72MTraining (log)
RiR19.239,527,7201,283.29MTraining (log)
Shake-Shake-ResNet-20-2x16d29.22546,93281.79MTraining (log)
Shake-Shake-ResNet-26-2x32d18.802,934,772428.90MTraining (log)
DIA-ResNet-2027.71292,71641.34MTraining (log)
DIA-ResNet-5624.35876,012127.18MTraining (log)
DIA-ResNet-11022.111,750,956255.95MTraining (log)
DIA-ResNet-164(BN)19.531,946,132259.20MTraining (log)
DIA-PreResNet-2028.37292,52441.32MTraining (log)
DIA-PreResNet-5625.05875,820127.16MTraining (log)
DIA-PreResNet-11022.691,750,764255.92MTraining (log)
DIA-PreResNet-164(BN)19.991,945,236258.97MTraining (log)

SVHN

ModelError, %ParamsFLOPs/2Remarks
NIN3.76966,986222.97MTraining (log)
ResNet-203.43272,47441.29MTraining (log)
ResNet-562.75855,770127.06MTraining (log)
ResNet-1102.451,730,714255.70MTraining (log)
ResNet-164(BN)2.421,704,154255.31MTraining (log)
ResNet-272(BN)2.432,816,986420.61MTraining (log)
ResNet-542(BN)2.345,599,066833.87MTraining (log)
ResNet-10012.4110,328,6021,536.40MTraining (log)
PreResNet-203.22272,28241.27MTraining (log)
PreResNet-562.80855,578127.03MTraining (log)
PreResNet-1102.791,730,522255.68MTraining (log)
PreResNet-164(BN)2.581,703,258255.08MTraining (log)
PreResNet-272(BN)2.342,816,090420.38MTraining (log)
PreResNet-542(BN)2.365,598,170833.64MTraining (log)
ResNeXt-20 (1x64d)2.983,446,602538.36MTraining (log)
ResNeXt-20 (2x32d)2.962,672,458425.15MTraining (log)
ResNeXt-20 (4x16d)3.172,285,386368.55MTraining (log)
ResNeXt-20 (8x8d)3.182,091,850340.25MTraining (log)
ResNeXt-20 (16x4d)3.211,995,082326.10MTraining (log)
ResNeXt-20 (32x2d)3.271,946,698319.03MTraining (log)
ResNeXt-20 (64x1d)3.421,922,506315.49MTraining (log)
ResNeXt-20 (2x64d)2.836,198,602987.98MTraining (log)
ResNeXt-20 (4x32d)2.984,650,314761.57MTraining (log)
ResNeXt-20 (8x16d)3.013,876,170648.37MTraining (log)
ResNeXt-20 (16x8d)2.933,489,098591.77MTraining (log)
ResNeXt-20 (32x4d)3.093,295,562563.47MTraining (log)
ResNeXt-20 (64x2d)3.143,198,794549.32MTraining (log)
ResNeXt-56 (1x64d)2.429,317,1941,399.33MTraining (log)
ResNeXt-56 (2x32d)2.466,994,7621,059.72MTraining (log)
ResNeXt-56 (4x16d)2.445,833,546889.91MTraining (log)
ResNeXt-56 (8x8d)2.475,252,938805.01MTraining (log)
ResNeXt-56 (16x4d)2.564,962,634762.56MTraining (log)
ResNeXt-56 (32x2d)2.534,817,482741.34MTraining (log)
ResNeXt-56 (64x1d)2.554,744,906730.72MTraining (log)
ResNeXt-29 (32x4d)2.804,775,754780.55MTraining (log)
ResNeXt-29 (16x64d)2.6868,155,21010,709.34MTraining (log)
ResNeXt-272 (1x64d)2.3544,540,7466,565.15MTraining (log)
ResNeXt-272 (2x32d)2.4432,928,5864,867.11MTraining (log)
SE-ResNet-203.23274,84741.30MTraining (log)
SE-ResNet-562.64862,889127.07MTraining (log)
SE-ResNet-1102.351,744,952255.72MTraining (log)
SE-ResNet-164(BN)2.451,906,258255.52MTraining (log)
SE-ResNet-272(BN)2.383,153,826420.96MTraining (log)
SE-ResNet-542(BN)2.266,272,746834.57MTraining (log)
SE-PreResNet-203.24274,55941.30MTraining (log)
SE-PreResNet-562.71862,601127.07MTraining (log)
SE-PreResNet-1102.591,744,664255.72MTraining (log)
SE-PreResNet-164(BN)2.561,904,882255.29MTraining (log)
SE-PreResNet-272(BN)2.493,152,450420.73MTraining (log)
SE-PreResNet-542(BN)2.476,271,370834.34MTraining (log)
PyramidNet-110 (a=48)2.471,772,706408.37MTraining (log)
PyramidNet-110 (a=84)2.433,904,446778.15MTraining (log)
PyramidNet-110 (a=270)2.3828,485,4774,730.60MTraining (log)
PyramidNet-164 (a=270, BN)2.3327,216,0214,608.81MTraining (log)
PyramidNet-200 (a=240, BN)2.3226,752,7024,563.40MTraining (log)
PyramidNet-236 (a=220, BN)2.3526,969,0464,631.32MTraining (log)
PyramidNet-272 (a=200, BN)2.4026,210,8424,541.36MTraining (log)
DenseNet-40 (k=12)3.05599,050210.80MTraining (log)
DenseNet-BC-40 (k=12)3.20176,12274.89MTraining (log)
DenseNet-BC-40 (k=24)2.90690,346293.09MTraining (log)
DenseNet-BC-40 (k=36)2.601,542,682654.60MTraining (log)
DenseNet-100 (k=12)2.604,068,4901,353.55MTraining (log)
X-DenseNet-BC-40-2 (k=24)2.87690,346293.09MTraining (log)
X-DenseNet-BC-40-2 (k=36)2.741,542,682654.60MTraining (log)
WRN-16-102.7817,116,6342,414.04MTraining (log)
WRN-28-102.7136,479,1945,246.98MTraining (log)
WRN-40-82.5435,748,3145,176.90MTraining (log)
WRN-20-10-1bit2.7326,737,1404,019.14MTraining (log)
WRN-20-10-32bit2.5926,737,1404,019.14MTraining (log)
RoR-3-562.69762,746113.43MTraining (log)
RoR-3-1102.571,637,690242.07MTraining (log)
RoR-3-1642.732,512,634370.72MTraining (log)
RiR2.689,492,9801,281.08MTraining (log)
Shake-Shake-ResNet-20-2x16d3.17541,08281.78MTraining (log)
Shake-Shake-ResNet-26-2x32d2.622,923,162428.89MTraining (log)
DIA-ResNet-203.23286,86641.34MTraining (log)
DIA-ResNet-562.68870,162127.18MTraining (log)
DIA-ResNet-1102.471,745,106255.94MTraining (log)
DIA-ResNet-164(BN)2.441,923,002259.18MTraining (log)
DIA-PreResNet-203.03286,67441.31MTraining (log)
DIA-PreResNet-562.80869,970127.15MTraining (log)
DIA-PreResNet-1102.421,744,914255.92MTraining (log)
DIA-PreResNet-164(BN)2.561,922,106258.95MTraining (log)

CUB-200-2011

ModelError, %ParamsFLOPs/2Remarks
ResNet-1027.655,008,392893.63MTraining (log)
ResNet-1226.585,082,3761,125.84MTraining (log)
ResNet-1424.355,377,8001,357.53MTraining (log)
ResNet-1623.216,558,4721,588.93MTraining (log)
ResNet-1823.3011,279,1121,820.00MTraining (log)
ResNet-2622.5217,549,8322,746.38MTraining (log)
SE-ResNet-1027.395,052,932893.67MTraining (log)
SE-ResNet-1226.045,127,4961,125.88MTraining (log)
SE-ResNet-1423.635,425,1041,357.58MTraining (log)
SE-ResNet-1623.216,614,2401,588.99MTraining (log)
SE-ResNet-1823.0811,368,1921,820.10MTraining (log)
SE-ResNet-2622.5117,683,4522,746.52MTraining (log)
MobileNet x1.023.463,411,976578.98MTraining (log)
ProxylessNAS Mobile21.883,055,712331.44MTraining (log)
NTS-Net13.2628,623,33333,361.39MFrom yangze0930/NTS-Net (log)

Pascal VOC20102

ModelExtractorPix.Acc.,%mIoU,%ParamsFLOPs/2Remarks
PSPNetResNet(D)-101b98.0981.4465,708,501230,586.69MFrom dmlc/gluon-cv (log)
DeepLabv3ResNet(D)-101b97.9580.2458,754,77347,624.54MFrom dmlc/gluon-cv (log)
DeepLabv3ResNet(D)-152b98.1181.2074,398,42159,894.06MFrom dmlc/gluon-cv (log)
FCN-8s(d)ResNet(D)-101b97.8080.4052,072,917196,562.96MFrom dmlc/gluon-cv (log)

ADE20K

ModelExtractorPix.Acc.,%mIoU,%ParamsFLOPs/2Remarks
PSPNetResNet(D)-50b79.3736.8746,782,550162,410.82MFrom dmlc/gluon-cv (log)
PSPNetResNet(D)-101b79.9337.9765,774,678230,824.47MFrom dmlc/gluon-cv (log)
DeepLabv3ResNet(D)-50b79.7237.1339,795,79832,755.38MFrom dmlc/gluon-cv (log)
DeepLabv3ResNet(D)-101b80.2137.8458,787,92647,650.43MFrom dmlc/gluon-cv (log)
FCN-8s(d)ResNet(D)-50b76.9233.3933,146,966128,387.08MFrom dmlc/gluon-cv (log)
FCN-8s(d)ResNet(D)-101b79.0135.8852,139,094196,800.73MFrom dmlc/gluon-cv (log)

Cityscapes

ModelExtractorPix.Acc.,%mIoU,%ParamsFLOPs/2Remarks
PSPNetResNet(D)-101b96.1771.7265,707,475230,583.01MFrom dmlc/gluon-cv (log)
ICNetResNet(D)-50b95.5064.0247,489,18414,241.91MFrom dmlc/gluon-cv (log)
Fast-SCNN-95.1465.761,138,0513,490.05MFrom dmlc/gluon-cv (log)
SINet-93.6660.31119,4181,411.97MFrom clovaai/c3_sinet (log)
DANetResNet(D)-50b95.9167.9947,586,427180,370.99MFrom dmlc/gluon-cv (log)
DANetResNet(D)-101b96.0368.1066,578,555248,784.64MFrom dmlc/gluon-cv (log)

COCO Semantic Segmentation

ModelExtractorPix.Acc.,%mIoU,%ParamsFLOPs/2Remarks
PSPNetResNet(D)-101b92.0567.4165,708,501230,586.69MFrom dmlc/gluon-cv (log)
DeepLabv3ResNet(D)-101b92.1967.7358,754,77347,624.54MFrom dmlc/gluon-cv (log)
DeepLabv3ResNet(D)-152b92.2468.9974,398,421275,084.22MFrom dmlc/gluon-cv (log)
FCN-8s(d)ResNet(D)-101b91.4460.1152,072,917196,562.96MFrom dmlc/gluon-cv (log)

CelebAMask-HQ

ModelExtractorParamsFLOPs/2Remarks
BiSeNetResNet-1813,300,416-From zllrunning/face...Torch (log)

COCO Keypoints Detection

ModelExtractorOKS AP, %ParamsFLOPs/2Remarks
AlphaPoseFast-SE-ResNet-101b74.15/91.59/80.6859,569,8739,553.15MFrom dmlc/gluon-cv (log)
SimplePoseResNet-1866.31/89.20/73.4115,376,7211,799.25MFrom dmlc/gluon-cv (log)
SimplePoseResNet-50b71.02/91.23/78.5733,999,6974,041.06MFrom dmlc/gluon-cv (log)
SimplePoseResNet-101b72.44/92.18/79.7652,991,8257,685.04MFrom dmlc/gluon-cv (log)
SimplePoseResNet-152b72.53/92.14/79.6168,635,47311,332.86MFrom dmlc/gluon-cv (log)
SimplePoseResNet(A)-50b71.70/91.31/78.6634,018,9294,278.56MFrom dmlc/gluon-cv (log)
SimplePoseResNet(A)-101b72.97/92.24/80.8153,011,0577,922.54MFrom dmlc/gluon-cv (log)
SimplePoseResNet(A)-152b73.44/92.27/80.7268,654,70511,570.36MFrom dmlc/gluon-cv (log)
SimplePose(Mobile)ResNet-1866.25/89.17/74.3212,858,2081,960.96MFrom dmlc/gluon-cv (log)
SimplePose(Mobile)ResNet-50b71.10/91.28/78.6725,582,9444,221.30MFrom dmlc/gluon-cv (log)
SimplePose(Mobile)1.0 MobileNet-22464.10/88.06/71.235,019,744751.36MFrom dmlc/gluon-cv (log)
SimplePose(Mobile)1.0 MobileNetV2b-22463.74/88.12/71.064,102,176495.95MFrom dmlc/gluon-cv (log)
SimplePose(Mobile)MobileNetV3 Small 224/1.054.34/83.67/59.352,625,088236.51MFrom dmlc/gluon-cv (log)
SimplePose(Mobile)MobileNetV3 Large 224/1.063.67/88.91/70.824,768,336403.97MFrom dmlc/gluon-cv (log)
Lightweight OpenPose 2DMobileNet39.99/65.95/40.704,091,6988,948.96MFrom Daniil-Osokin/lighw...ch (log)
Lightweight OpenPose 3DMobileNet39.99/65.95/40.705,085,98311,049.43MFrom Daniil-Osokin/li...3d...ch (log)
IBPPose-64.86/83.62/70.1295,827,78457,193.82MFrom jialee93/Improved...Parts (log)

Mozilla Common Voice (Corpus 6.1, dev subset)

Some remarks:

  • NR means Noise Reduction.
  • LS means trained on LibriSpeech dataset.
ModelLangWER, %ParamsFLOPs/2Remarks
Jasper DR 10x5En21.90332,632,34985,142.96MFrom NVIDIA/NeMo (log)
Jasper DR 10x5 NREn17.89332,632,34985,142.96MFrom NVIDIA/NeMo (log)
QuartzNet 5x5 LSEn44.686,713,1811,717.12MFrom NVIDIA/NeMo (log)
QuartzNet 15x5En16.7718,924,3814,840.29MFrom NVIDIA/NeMo (log)
QuartzNet 15x5 NREn17.7418,924,3814,840.29MFrom NVIDIA/NeMo (log)
QuartzNet 15x5De11.6618,927,4564,841.08MFrom NVIDIA/NeMo (log)
QuartzNet 15x5Fr13.8818,938,7314,843.96MFrom NVIDIA/NeMo (log)
QuartzNet 15x5It15.0218,934,6314,842.91MFrom NVIDIA/NeMo (log)
QuartzNet 15x5Es12.9518,931,5564,842.13MFrom NVIDIA/NeMo (log)
QuartzNet 15x5Ca8.4218,934,6314,842.91MFrom NVIDIA/NeMo (log)
QuartzNet 15x5Pl13.5918,929,5064,841.60MFrom NVIDIA/NeMo (log)
QuartzNet 15x5Ru16.4818,930,5314,841.87MFrom NVIDIA/NeMo (log)
QuartzNet 15x5Ru/349.6818,929,5064,841.60MFrom sberdevices/golos (log)

Keywords

FAQs


Did you know?

Socket

Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.

Install

Related posts

SocketSocket SOC 2 Logo

Product

  • Package Alerts
  • Integrations
  • Docs
  • Pricing
  • FAQ
  • Roadmap
  • Changelog

Packages

npm

Stay in touch

Get open source security insights delivered straight into your inbox.


  • Terms
  • Privacy
  • Security

Made with ⚡️ by Socket Inc