diff --git a/model_zoo/official/README.md b/model_zoo/official/README.md index 8aaf07071a7446b4c69850af0ae42e5f6579ac47..2351d8d11385e6d16259ae2ca5ac39dac5b54d6b 100644 --- a/model_zoo/official/README.md +++ b/model_zoo/official/README.md @@ -24,9 +24,9 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework, - [ResNet101](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnet/README.md) - [ResNext50](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnext50/README.md) - [VGG16](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/vgg16/README.md) - - [AlexNet](#https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/alexnet/README.md) + - [AlexNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/alexnet/README.md) - [LeNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/lenet/README.md) - - [LeNet](#https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/lenet_quant/README.md) + - [LeNet_Quant](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/lenet_quant/README.md) - [Object Detection and Segmentation](#object-detection-and-segmentation) - [DeepLabV3](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/deeplabv3/README.md) - [FasterRCNN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/faster_rcnn/README.md) @@ -36,7 +36,7 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework, - [MobileNetV2_Quant](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/mobilenetv2_quant/README.md) - [MobileNetV3](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/mobilenetv3/README.md) - [SSD](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/ssd/README.md) - - [Warp-CTC](#https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/warpctc/README.md) + - [Warp-CTC](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/warpctc/README.md) - [Natural Language Processing](#natural-language-processing) - [BERT[benchmark]](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/bert/README.md) @@ -44,11 +44,11 @@ In order to facilitate developers to enjoy the benefits of MindSpore framework, - [MASS](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/mass/README.md) - [Transformer](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/transformer/README.md) - [Recommendation](#recommendation) - - [DeepFM](#https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/recommend/deepfm/README.md) + - [DeepFM](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/recommend/deepfm/README.md) - [Wide&Deep[benchmark]](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/recommend/wide_and_deep/README.md) - [Graph Neural Networks](#gnn) - - [GAT](#https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/gnn/gat/README.md) - - [GCN](#https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/gnn/gcn//README.md) + - [GAT](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/gnn/gat/README.md) + - [GCN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/gnn/gcn//README.md) diff --git a/model_zoo/official/cv/googlenet/mindspore_hub_conf.py b/model_zoo/official/cv/googlenet/mindspore_hub_conf.py new file mode 100644 index 0000000000000000000000000000000000000000..600838c470286210b59e181987800803fbe931b3 --- /dev/null +++ b/model_zoo/official/cv/googlenet/mindspore_hub_conf.py @@ -0,0 +1,25 @@ +# Copyright 2020 Huawei Technologies Co., Ltd +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============================================================================ +"""hub config.""" +from src.googlenet import GoogleNet + +def googlenet(*args, **kwargs): + return GoogleNet(*args, **kwargs) + + +def create_network(name, *args, **kwargs): + if name == "googlenet": + return googlenet(*args, **kwargs) + raise NotImplementedError(f"{name} is not implemented in the repo")