diff --git a/core/sdk-cpp/include/endpoint_config.h b/core/sdk-cpp/include/endpoint_config.h index 6edb6ed8ab3b7c62be12c35c1658a2adf3140341..f814b659a24c4e5cb6e352c9a39bcfef1df147c1 100644 --- a/core/sdk-cpp/include/endpoint_config.h +++ b/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 diff --git a/python/examples/unet_for_image_seg/seg_client.py b/python/examples/unet_for_image_seg/seg_client.py index d9494508dba869ecb34c596520ec6edbefaf0bbf..9e76b060955ec74492312c8896efaf3946a3f7ab 100644 --- a/python/examples/unet_for_image_seg/seg_client.py +++ b/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) diff --git a/python/paddle_serving_app/models/model_list.py b/python/paddle_serving_app/models/model_list.py index 594b303072079fba0fee6224f86e485caf88d776..c2638f307ba5d51f95e84846f1e9a5995bcf104e 100644 --- a/python/paddle_serving_app/models/model_list.py +++ b/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/"