fix unet demo

ca6e8dc2 · MRXLT · ddd533e2 · ca6e8dc2
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

python/examples/unet_for_image_seg/seg_client.py python/examples/unet_for_image_seg/seg_client.py +2 -1

未找到文件。
--- 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)