From defb8fea4ca32bdd2608428e551e67f600450c18 Mon Sep 17 00:00:00 2001 From: Tong He Date: Tue, 1 May 2018 20:26:27 -0700 Subject: [PATCH] add mobilenet v2 1.0 pretrained model (#10774) * add mobilenet v2 1.0 pretrained model * add auto summary --- docs/api/python/gluon/model_zoo.md | 3 +++ python/mxnet/gluon/model_zoo/model_store.py | 1 + 2 files changed, 4 insertions(+) diff --git a/docs/api/python/gluon/model_zoo.md b/docs/api/python/gluon/model_zoo.md index cb12faba8506..950e2c02c2dc 100644 --- a/docs/api/python/gluon/model_zoo.md +++ b/docs/api/python/gluon/model_zoo.md @@ -42,6 +42,7 @@ The following table summarizes the available models. | mobilenet0.5 | [MobileNet 0.5](https://arxiv.org/abs/1704.04861) | 1,342,536 | 0.6307 | 0.8475 | Trained with [script](https://github.com/apache/incubator-mxnet/blob/master/example/gluon/image_classification.py) | | mobilenet0.75 | [MobileNet 0.75](https://arxiv.org/abs/1704.04861) | 2,601,976 | 0.6738 | 0.8782 | Trained with [script](https://github.com/apache/incubator-mxnet/blob/master/example/gluon/image_classification.py) | | mobilenet1.0 | [MobileNet 1.0](https://arxiv.org/abs/1704.04861) | 4,253,864 | 0.7105 | 0.9006 | Trained with [script](https://github.com/apache/incubator-mxnet/blob/master/example/gluon/image_classification.py) | +| mobilenetv2_1.0 | [MobileNetV2 1.0](https://arxiv.org/abs/1801.04381) | 3,539,136 | 0.7159 | 0.9047 | Trained with [script](https://github.com/dmlc/gluon-cv/blob/15ed8a4c71d411b878f0d71d1c7afdce6710c913/scripts/classification/imagenet/train_imagenet.py) | | resnet18_v1 | [ResNet-18 V1](http://arxiv.org/abs/1512.03385) | 11,699,112 | 0.7039 | 0.8959 | Trained with [script](https://github.com/dmlc/gluon-cv/blob/15ed8a4c71d411b878f0d71d1c7afdce6710c913/scripts/classification/imagenet/train_imagenet.py) | | resnet34_v1 | [ResNet-34 V1](http://arxiv.org/abs/1512.03385) | 21,814,696 | 0.7411 | 0.9184 | Trained with [script](https://github.com/dmlc/gluon-cv/blob/15ed8a4c71d411b878f0d71d1c7afdce6710c913/scripts/classification/imagenet/train_imagenet.py) | | resnet50_v1 | [ResNet-50 V1](http://arxiv.org/abs/1512.03385) | 25,629,032 | 0.7540 | 0.9266 | Trained with [script](https://github.com/apache/incubator-mxnet/blob/master/example/gluon/image_classification.py) | @@ -203,6 +204,7 @@ The following table summarizes the available models. mobilenet0_75 mobilenet0_5 mobilenet0_25 + mobilenet_v2_1_0 ``` ```eval_rst @@ -210,6 +212,7 @@ The following table summarizes the available models. :nosignatures: MobileNet + MobileNetV2 ``` ## API Reference diff --git a/python/mxnet/gluon/model_zoo/model_store.py b/python/mxnet/gluon/model_zoo/model_store.py index 6dc57acd1eed..14ec8d0a2366 100644 --- a/python/mxnet/gluon/model_zoo/model_store.py +++ b/python/mxnet/gluon/model_zoo/model_store.py @@ -35,6 +35,7 @@ ('8e9d539cc66aa5efa71c4b6af983b936ab8701c3', 'mobilenet0.5'), ('529b2c7f4934e6cb851155b22c96c9ab0a7c4dc2', 'mobilenet0.75'), ('6b8c5106c730e8750bcd82ceb75220a3351157cd', 'mobilenet1.0'), + ('3ab4967b7a12a9246a144c9dfff74506cb78a526', 'mobilenetv2_1.0'), ('e54b379f50fa4b10bbd2506237e3bd74e6164778', 'resnet18_v1'), ('c1dc0967a3d25ee9127e03bc1046a5d44d92e2ba', 'resnet34_v1'), ('c940b1a062b32e3a5762f397c9d1e178b5abd007', 'resnet50_v1'),