From ec7e73f2821168c931c8e50e658d7c7ce7215a08 Mon Sep 17 00:00:00 2001 From: chenjian Date: Tue, 7 Dec 2021 14:23:43 +0800 Subject: [PATCH] fix U2Net_Portrait module readme (#1708) --- .../image/Image_gan/style_transfer/U2Net_Portrait/README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/image/Image_gan/style_transfer/U2Net_Portrait/README.md b/modules/image/Image_gan/style_transfer/U2Net_Portrait/README.md index e2724618..4175ec59 100644 --- a/modules/image/Image_gan/style_transfer/U2Net_Portrait/README.md +++ b/modules/image/Image_gan/style_transfer/U2Net_Portrait/README.md @@ -50,16 +50,16 @@ ## 三、模型API预测 -- ### 1、代码示例 +- ### 1、预测代码示例 - ```python import paddlehub as hub import cv2 model = hub.Module(name="U2Net_Portrait") - result = model.Cartoon_GEN(images=[cv2.imread('/PATH/TO/IMAGE')]) + result = model.Portrait_GEN(images=[cv2.imread('/PATH/TO/IMAGE')]) # or - # result = model.Cartoon_GEN(paths=['/PATH/TO/IMAGE']) + # result = model.Portrait_GEN(paths=['/PATH/TO/IMAGE']) ``` - ### 2、API -- GitLab