From 2e0d1d8b43747ccd07ae1d402dadc05dbdd7d2ea Mon Sep 17 00:00:00 2001 From: chenjian Date: Tue, 8 Mar 2022 18:54:38 +0800 Subject: [PATCH] fix according to review --- .../style_transfer/lapstyle_starrynew/module.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/modules/image/Image_gan/style_transfer/lapstyle_starrynew/module.py b/modules/image/Image_gan/style_transfer/lapstyle_starrynew/module.py index e175eb2a..fffba04b 100644 --- a/modules/image/Image_gan/style_transfer/lapstyle_starrynew/module.py +++ b/modules/image/Image_gan/style_transfer/lapstyle_starrynew/module.py @@ -52,12 +52,12 @@ class Lapstyle_starrynew: ''' Transfer a image to starrynew style. - images (list[dict]): data of images, 每一个元素都为一个 dict,有关键字 content, style, 相应取值为: - - content (numpy.ndarray): 待转换的图片,shape 为 \[H, W, C\],BGR格式;
- - style (numpy.ndarray) : 风格图像,shape为 \[H, W, C\],BGR格式;
- paths (list[str]): paths to images, 每一个元素都为一个dict, 有关键字 content, style, 相应取值为: - - content (str): 待转换的图片的路径;
- - style (str) : 风格图像的路径;
+ images (list[dict]): data of images, each element is a dict: + - content (numpy.ndarray): input image,shape is \[H, W, C\],BGR format;
+ - style (numpy.ndarray) : style image,shape is \[H, W, C\],BGR format;
+ paths (list[str]): paths to images, eacg element is a dict: + - content (str): path to input image;
+ - style (str) : path to style image;
output_dir: the dir to save the results use_gpu: if True, use gpu to perform the computation, otherwise cpu. -- GitLab