瀏覽代碼

依赖版本

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):