Upload 2 files

demoire_noAug.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:398e8036ee88a177f0d3f7006b47040e9838375f55a75ddc8efdc68f86318710
+size 89262019

model.py

@@ -142,7 +142,7 @@ def init():
         dehaze_model = FPro(**dehaze_cfg)
 
         # 加载权重
-        dehaze_ckpt = torch.load("dehaze.pth", map_location='cpu')
+        dehaze_ckpt = torch.load("synDehaze.pth", map_location='cpu')
         dehaze_model.load_state_dict(dehaze_ckpt['params'])
         dehaze_model.to(device)
         dehaze_model.eval()
@@ -180,7 +180,7 @@ def init():
             }
 
         demoiring_model = FPro(**demoiring_cfg)
-        demoiring_ckpt = torch.load("deblur.pth", map_location='cpu')
+        demoiring_ckpt = torch.load("demoire_noAug.pth", map_location='cpu')
         demoiring_model.load_state_dict(demoiring_ckpt['params'])
         demoiring_model.to(device)
         demoiring_model.eval()