mmclassification/docs/model_zoo.md

6.1 KiB

Model Zoo

ImageNet

ImageNet has multiple versions, but the most commonly used one is ILSVRC 2012. The ResNet family models below are trained by standard data augmentations, i.e., RandomResizedCrop, RandomHorizontalFlip and Normalize.

Model Params(M) Flops(G) Top-1 (%) Top-5 (%) Download
ResNet-18 11.69 1.82 70.07 89.44 model | log
ResNet-34 21.8 3.68 73.85 91.53 model | log
ResNet-50 25.56 4.12 76.55 93.15 model | log
ResNet-101 44.55 7.85 78.18 94.03 model | log
ResNet-152 60.19 11.58 78.63 94.16 model | log
ResNetV1D-50 25.58 4.36 77.4 93.66 model | log
ResNetV1D-101 44.57 8.09 78.85 94.38 model | log
ResNetV1D-152 60.21 11.82 79.35 94.61 model | log
ResNeXt-32x4d-50 25.03 4.27 77.92 93.74 model | log
ResNeXt-32x4d-101 44.18 8.03 78.7 94.34 model | log
ResNeXt-32x8d-101 88.79 16.5 79.22 94.52 model | log
ResNeXt-32x4d-152 59.95 11.8 79.06 94.47 model | log
SE-ResNet-50 28.09 4.13 77.74 93.84 model | log
SE-ResNet-101 49.33 7.86 78.26 94.07 model | log
ShuffleNetV1 1.0x (group=3) 1.87 0.146 68.13 87.81 model | log
ShuffleNetV2 1.0x 2.28 0.149 69.55 88.92 model | log
MobileNet V2 3.5 0.319 71.86 90.42 model | log

Models with * are converted from other repos, others are trained by ourselves.