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 e175eb2aedc56b486b0b9f7ebf3ca253835f6693..fffba04b06f2d612b0e197c008a8c92bc982c598 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.