fix input of unet

core/sdk-cpp/include/endpoint_config.h

@@ -22,23 +22,23 @@ namespace baidu {
 namespace paddle_serving {
 namespace sdk_cpp {
 
-#define PARSE_CONF_ITEM(conf, item, name, fail)             \
-  do {                                                      \
-    if (conf.has_##name()) {                                \
-      item.set(conf.name());                                \
-    } else {                                                \
-      LOG(ERROR) << "Not found key in configue: " << #name; \
-    }                                                       \
+#define PARSE_CONF_ITEM(conf, item, name, fail)          \
+  do {                                                   \
+    if (conf.has_##name()) {                             \
+      item.set(conf.name());                             \
+    } else {                                             \
+      VLOG(2) << "Not found key in configue: " << #name; \
+    }                                                    \
   } while (0)
 
-#define ASSIGN_CONF_ITEM(dest, src, fail)                          \
-  do {                                                             \
-    if (!src.init) {                                               \
-      LOG(ERROR) << "Cannot assign an unintialized item: " << #src \
-                 << " to dest: " << #dest;                         \
-      return fail;                                                 \
-    }                                                              \
-    dest = src.value;                                              \
+#define ASSIGN_CONF_ITEM(dest, src, fail)                       \
+  do {                                                          \
+    if (!src.init) {                                            \
+      VLOG(2) << "Cannot assign an unintialized item: " << #src \
+              << " to dest: " << #dest;                         \
+      return fail;                                              \
+    }                                                           \
+    dest = src.value;                                           \
   } while (0)
 
 template <typename T>

python/examples/unet_for_image_seg/seg_client.py

@@ -18,7 +18,7 @@ import sys
 import cv2
 
 client = Client()
-client.load_client_config(sys.argv[1])
+client.load_client_config("unet_client/serving_client_conf.prototxt")
 client.connect(["127.0.0.1:9494"])
 
 preprocess = Sequential(
@@ -27,9 +27,7 @@ preprocess = Sequential(
 
 postprocess = SegPostprocess(2)
 
-for i in range(100):
-    filename = sys.argv[2]
-    im = preprocess(filename)
-    fetch_map = client.predict(feed={"image": im}, fetch=["output"])
-    fetch_map["filename"] = filename
-    postprocess(fetch_map)
+im = preprocess("N0060.jpg")
+fetch_map = client.predict(feed={"image": im}, fetch=["output"])
+fetch_map["filename"] = filename
+postprocess(fetch_map)

python/paddle_serving_app/models/model_list.py

@@ -27,7 +27,7 @@ class ServingModels(object):
         self.model_dict["ObjectDetection"] = ["faster_rcnn", "yolov3"]
         self.model_dict["ImageSegmentation"] = ["unet", "deeplabv3"]
         self.model_dict["ImageClassification"] = [
-            "resnet_v2_50_imagenet", "efficientnetb6_imagenet"
+            "resnet_v2_50_imagenet", "mobilenet_v2_imagenet"
         ]
 
         image_class_url = "https://paddle-serving.bj.bcebos.com/paddle_hub_models/image/ImageClassification/"