소스 검색

依赖版本

tuonian 1 년 전
부모
커밋
b96eab93b7
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      tools/model.py

+ 2 - 2
tools/model.py

@@ -21,8 +21,8 @@ def get_rel_path(path: str):
 class MattingModel:
     path = ""
     config = ""
-    model: paddle.nn.layer.Layer
-    transforms: Compose
+    model = None
+    transforms: ppmatting.transforms.Compose
     init = False
 
     def __init__(self, p, c):