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 41012b9614488ec9f807f4e9acb4709abbda264a..d657fb88ba9b6c8c6f30856caf2bb9b440b04a90 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.