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 e2724618d42095ebf05e410ece2ced9b06c831d6..4175ec598c6e02a65a744a9b26dd7c00aa2efd43 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