add fasterrcnn, mass, unet, psenet hub conf file.

model_zoo/official/cv/faster_rcnn/mindspore_hub_conf.py

@@ -0,0 +1,22 @@
+# 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.FasterRcnn.faster_rcnn_r50 import Faster_Rcnn_Resnet50
+from src.config import config
+
+def create_network(name, *args, **kwargs):
+    if name == "faster_rcnn":
+        return Faster_Rcnn_Resnet50(config=config)
+    raise NotImplementedError(f"{name} is not implemented in the repo")

model_zoo/official/cv/psenet/mindspore_hub_conf.py

@@ -0,0 +1,24 @@
+# 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.ETSNET.etsnet import ETSNet
+from src.config import config
+
+def create_network(name, *args, **kwargs):
+    if name == "psenet":
+        infer_mode = kwargs.get("infer_mode", False)
+        config.INFERENCE = infer_mode
+        return ETSNet(config)
+    raise NotImplementedError(f"{name} is not implemented in the repo")

model_zoo/official/cv/unet/mindspore_hub_conf.py

@@ -0,0 +1,22 @@
+# 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.unet import UNet
+
+def create_network(name, *args, **kwargs):
+    if name == "unet2d":
+        unet = UNet(*args, **kwargs)
+        return unet
+    raise NotImplementedError(f"{name} is not implemented in the repo")

model_zoo/official/nlp/mass/mindspore_hub_conf.py

@@ -0,0 +1,38 @@
+# 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."""
+import mindspore.common.dtype as mstype
+
+from config import TransformerConfig
+from src.transformer import TransformerNetworkWithLoss, TransformerInferModel
+
+def get_config(config):
+    config = TransformerConfig.from_json_file(config)
+    config.compute_type = mstype.float16
+    config.dtype = mstype.float32
+    return config
+
+def create_network(name, *args, **kwargs):
+    """create mass network."""
+    if name == "mass":
+        if "config" in kwargs:
+            config = get_config(kwargs["config"])
+        else:
+            raise NotImplementedError(f"Please make sure the configuration file path is correct")
+        is_training = kwargs.get("is_training", False)
+        if is_training:
+            return TransformerNetworkWithLoss(config, is_training=is_training, *args)
+        return TransformerInferModel(config, *args)
+    raise NotImplementedError(f"{name} is not implemented in the repo")