From ad2f39f4a6748ac5ffeacaa74154d70d72c5f095 Mon Sep 17 00:00:00 2001 From: MRXLT Date: Mon, 1 Jun 2020 12:07:15 +0000 Subject: [PATCH] fix unet demo --- python/examples/unet_for_image_seg/seg_client.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/examples/unet_for_image_seg/seg_client.py b/python/examples/unet_for_image_seg/seg_client.py index 9e76b060..44f634b6 100644 --- a/python/examples/unet_for_image_seg/seg_client.py +++ b/python/examples/unet_for_image_seg/seg_client.py @@ -27,7 +27,8 @@ preprocess = Sequential( postprocess = SegPostprocess(2) -im = preprocess("N0060.jpg") +filename = "N0060.jpg" +im = preprocess(filename) fetch_map = client.predict(feed={"image": im}, fetch=["output"]) fetch_map["filename"] = filename postprocess(fetch_map) -- GitLab