From 8367e974fcda30b46274257133eae4786837829c Mon Sep 17 00:00:00 2001
From: ShenYuhan <yhan_shen@163.com>
Date: Tue, 18 Feb 2020 12:31:35 +0800
Subject: [PATCH] modify visual_result

---
 paddlehub/serving/app_single.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlehub/serving/app_single.py b/paddlehub/serving/app_single.py
index 0b05af4a..1cfeb94f 100644
--- a/paddlehub/serving/app_single.py
+++ b/paddlehub/serving/app_single.py
@@ -147,7 +147,7 @@ def predict_mask(module, input_img, id, batch_size, extra=None, r_img=True):
             data.update(input_img)
         if extra is not None:
             data.update(extra)
-            r_img = False if "r_img" in extra.keys() else True
+            r_img = False if "visual_result" in extra.keys() else True
         results = predict_method(
             data=data,
             visualization=r_img,
-- 
GitLab