From 3f7c2874b2f75b11d4f9c033ff7512d8bf139ae7 Mon Sep 17 00:00:00 2001 From: chenjian Date: Tue, 8 Mar 2022 18:53:59 +0800 Subject: [PATCH] fix according to review --- .../style_transfer/lapstyle_circuit/module.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/modules/image/Image_gan/style_transfer/lapstyle_circuit/module.py b/modules/image/Image_gan/style_transfer/lapstyle_circuit/module.py index 41012b96..d657fb88 100644 --- a/modules/image/Image_gan/style_transfer/lapstyle_circuit/module.py +++ b/modules/image/Image_gan/style_transfer/lapstyle_circuit/module.py @@ -52,12 +52,16 @@ class Lapstyle_circuit: ''' Transfer a image to circuit 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. + visualization: if True, save results in output_dir. output_dir: the dir to save the results use_gpu: if True, use gpu to perform the computation, otherwise cpu. -- GitLab