From b84b324cd34d23a4dd69726e0f2444c072fe273c Mon Sep 17 00:00:00 2001 From: chenjian Date: Mon, 14 Mar 2022 10:57:23 +0800 Subject: [PATCH] add declaration --- modules/image/Image_gan/gan/photopen/module.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/modules/image/Image_gan/gan/photopen/module.py b/modules/image/Image_gan/gan/photopen/module.py index 4ff21ceb..f8a23e57 100644 --- a/modules/image/Image_gan/gan/photopen/module.py +++ b/modules/image/Image_gan/gan/photopen/module.py @@ -40,17 +40,18 @@ class Photopen: self.network = PhotoPenPredictor(weight_path=self.pretrained_model, gen_cfg=cfg.predict) def photo_transfer(self, - images=None, - paths=None, - output_dir='./transfer_result/', - use_gpu=False, - visualization=True): + images: list = None, + paths: list = None, + output_dir: str = './transfer_result/', + use_gpu: bool = False, + visualization: bool = True): ''' images (list[numpy.ndarray]): data of images, shape of each is [H, W, C], color space must be BGR(read by cv2). paths (list[str]): paths to images - 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 (str): the dir to save the results + use_gpu (bool): if True, use gpu to perform the computation, otherwise cpu. + visualization (bool): if True, save results in output_dir. ''' results = [] paddle.disable_static() -- GitLab