Spaces:

Andy1621
/

IAT_enhancement

Build error

SakuraD commited on Oct 15, 2022

Commit

cabb20f

•

1 Parent(s): dd3ec97

init

Files changed (3) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ def dark_inference(img):
     transform = Compose([
         ToTensor(),
-        Scale(512),
         Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
         ConvertImageDtype(torch.float)
     ])
@@ -48,7 +48,7 @@ def exposure_inference(img):
     transform = Compose([
         ToTensor(),
-        Scale(512),
         ConvertImageDtype(torch.float)
     ])
     input_img = transform(img)

     transform = Compose([
         ToTensor(),
+        Scale(384),
         Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
         ConvertImageDtype(torch.float)
     ])
     transform = Compose([
         ToTensor(),
+        Scale(384),
         ConvertImageDtype(torch.float)
     ])
     input_img = transform(img)

test_dark.ipynb CHANGED Viewed

The diff for this file is too large to render. See raw diff

test_exposure.ipynb CHANGED Viewed

The diff for this file is too large to render. See raw diff