提交 93c83ac6 编写于 作者: W wangjiawei04

add mobilenet_cityspaces for image sg

上级 a1a8e32c
...@@ -25,7 +25,9 @@ class ServingModels(object): ...@@ -25,7 +25,9 @@ class ServingModels(object):
self.model_dict["SemanticRepresentation"] = ["ernie_base"] self.model_dict["SemanticRepresentation"] = ["ernie_base"]
self.model_dict["ChineseWordSegmentation"] = ["lac"] self.model_dict["ChineseWordSegmentation"] = ["lac"]
self.model_dict["ObjectDetection"] = ["faster_rcnn", "yolov3"] self.model_dict["ObjectDetection"] = ["faster_rcnn", "yolov3"]
self.model_dict["ImageSegmentation"] = ["unet", "deeplabv3"] self.model_dict["ImageSegmentation"] = [
"unet", "deeplabv3", "mobilenet_cityspaces"
]
self.model_dict["ImageClassification"] = [ self.model_dict["ImageClassification"] = [
"resnet_v2_50_imagenet", "mobilenet_v2_imagenet" "resnet_v2_50_imagenet", "mobilenet_v2_imagenet"
] ]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册