add declaration
@@ -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()
