Browse Source

依赖版本

tuonian 1 year ago
parent
commit
b96eab93b7
1 changed files with 2 additions and 2 deletions
  1. 2 2
      tools/model.py

+ 2 - 2
tools/model.py

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