README.md

ShuffleNet Series

ShuffleNet Series by Megvii Research.

Introduction

This repository contains the following ShuffleNet series models:

• ShuffleNetV1: ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices
• ShuffleNetV2: ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design
• ShuffleNetV2+: A strengthen version of ShuffleNetV2.
• ShuffleNetV2.Large: A deeper version based on ShuffleNetV2.
• OneShot: Single Path One-Shot Neural Architecture Search with Uniform Sampling
• DetNAS: DetNAS: Backbone Search for Object Detection

Trained Models

OneDrive download: Link

Details

ShuffleNetV2+

The following is the comparison between ShuffleNetV2+ and MobileNetV3. Details can be seen in ShuffleNetV2+.

Model FLOPs #Params Top-1 Top-5 ShuffleNetV2+ Large 360M 6.7M 22.9 6.7 MobileNetV3 Large 224/1.25 356M 7.5M 23.4 - ShuffleNetV2+ Medium 222M 5.6M 24.3 7.4 MobileNetV3 Large 224/1.0 217M 5.4M 24.8 - ShuffleNetV2+ Small 156M 5.1M 25.9 8.3 MobileNetV3 Large 224/0.75 155M 4.0M 26.7 -

ShuffleNetV2

The following is the comparison between ShuffleNetV2 and MobileNetV2. Details can be seen in ShuffleNetV2.

Model FLOPs #Params Top-1 Top-5 ShuffleNetV2 2.0x 591M 7.4M 25.0 7.6 MobileNetV2 (1.4) 585M 6.9M 25.3 - ShuffleNetV2 1.5x 299M 3.5M 27.4 9.4 MobileNetV2 300M 3.4M 28.0 - ShuffleNetV2 1.0x 146M 2.3M 30.6 11.1 ShuffleNetV2 0.5x 41M 1.4M 38.9 17.4

ShuffleNetV2.Large

The following is the comparison between ShuffleNetV2.Large and SENet. Details can be seen in ShuffleNetV2.Large.

Model FLOPs #Params Top-1 Top-5 ShuffleNetV2.Large 12.7G 140.7M 18.56 4.48 SENet 20.7G - 18.68 4.47

ShuffleNetV1

The following is the comparison between ShuffleNetV1 and MobileNetV1. Details can be seen in ShuffleNetV1.

Model FLOPs #Params Top-1 Top-5 ShuffleNetV1 2.0x (group=3) 524M 5.4M 25.9 8.6 ShuffleNetV1 2.0x (group=8) 522M 6.5M 27.1 9.2 1.0 MobileNetV1-224 569M 4.2M 29.4 - ShuffleNetV1 1.5x (group=3) 292M 3.4M 28.4 9.8 ShuffleNetV1 1.5x (group=8) 290M 4.3M 29.0 10.4 0.75 MobileNetV1-224 325M 2.6M 31.6 - ShuffleNetV1 1.0x (group=3) 138M 1.9M 32.2 12.3 ShuffleNetV1 1.0x (group=8) 138M 2.4M 32.0 13.6 0.5 MobileNetV1-224 149M 1.3M 36.3 - ShuffleNetV1 0.5x (group=3) 38M 0.7M 42.7 20.0 ShuffleNetV1 0.5x (group=8) 40M 1.0M 41.2 19.0 0.25 MobileNetV1-224 41M 0.5M 49.4 -

OneShot

The following is the comparison between Single Path One-Shot NAS and other NAS counterparts. Details can be seen in OneShot.

Model FLOPs #Params Top-1 Top-5 OneShot 328M 3.4M 25.1 8.0 NASNET-A 564M 5.3M 26.0 8.4 PNASNET 588M 5.1M 25.8 8.1 MnasNet 317M 4.2M 26.0 8.2 DARTS 574M 4.7M 26.7 8.7 FBNet-B 295M 4.5M 25.9 -

DetNAS

The following is the performance of DetNAS on ImageNet, compared with ResNet. Details can be seen in DetNAS.

Model FLOPs #Params Top-1 Top-5 mAP (COCO, FPN)* DetNAS_small 300M 3.7M 25.9 8.3 36.4 DetNAS_medium 1.3G 10.4M 22.8 6.5 40.0 DetNAS_large 3.8G 29.5M 21.6 6.3 42.0 ResNet50 3.8G - 23.9 7.1 37.3 ResNet101 7.6G - 22.6 6.4 40.0

*COCO models are coming soon.