fix

modules/image/Image_gan/style_transfer/face_parse/README.md

@@ -4,7 +4,7 @@
 | :--- | :---: |
 |类别|图像 - 人脸解析|
 |网络|BiSeNet|
-|数据集|-|
+|数据集|COCO-Stuff|
 |是否支持Fine-tuning|否|
 |模型大小|77MB|
 |最新更新日期|2021-12-07|

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

@@ -37,19 +37,19 @@ class Face_parse:
         self.network = FaceParsePredictor()
 
     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):
         '''
 
 
         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()