add type declaration

modules/image/Image_gan/style_transfer/lapstyle_stars/module.py

@@ -44,24 +44,24 @@ class Lapstyle_stars:
         self.network = LapStylePredictor(weight_path=self.pretrained_model)
 
     def style_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):
         '''
         Transfer a image to stars style.
 
         images (list[dict]): data of images, each element is a dict:
           - content (numpy.ndarray): input image，shape is \[H, W, C\]，BGR format；<br/>
           - style (numpy.ndarray) : style image，shape is \[H, W, C\]，BGR format；<br/>
-        paths (list[str]): paths to images, eacg element is a dict:
+        paths (list[dict]): paths to images, eacg element is a dict:
           - content (str): path to input image；<br/>
           - style (str) : path to style image；<br/>
 
-        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()