Spaces:

bingbingbing
/

ImageEnhancement

Runtime error

chenzhicun commited on May 19, 2022

Commit

cda9597

•

1 Parent(s): 96cea77

初始化web demo.

Files changed (4) hide show

__pycache__/torchvision_x_functional.cpython-310.pyc ADDED Viewed

Binary file (15.3 kB). View file

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ if cuda:
     classifier = classifier.cuda()
 # Load pretrained models
-cache = download_cached_file('https://drive.google.com/uc?export=download&id=1tzeECo1m4MBqvfLv4H4SQ7by4YMEP17H',
                              check_hash=False, progress=True)
 LUTs = torch.load(cache, map_location=torch.device('cpu'))
 LUT0.load_state_dict(LUTs["0"])
@@ -37,7 +37,7 @@ LUT0.eval()
 LUT1.eval()
 LUT2.eval()
-cache = download_cached_file('https://drive.google.com/uc?export=download&id=1rQ_p3NMRFxZ52MOYj0jPewYtD3JQTJGi',
                              check_hash=False, progress=True)
 classifier.load_state_dict(torch.load(cache, map_location=torch.device('cpu')))
 classifier.eval()
@@ -55,7 +55,7 @@ if cuda:
     Xclassifier = Xclassifier.cuda()
 # Load pretrained models
-cache = download_cached_file('https://drive.google.com/uc?export=download&id=1ossTzgbgpZL4Jy5uhiRJDGfCWw9vOv0c',
                              check_hash=False, progress=True)
 XLUTs = torch.load(cache, map_location=torch.device('cpu'))
 XLUT0.load_state_dict(XLUTs["0"])
@@ -65,7 +65,7 @@ XLUT0.eval()
 XLUT1.eval()
 XLUT2.eval()
-cache = download_cached_file('https://drive.google.com/uc?export=download&id=1279CoaqQZK-eK83283MERoRxtRbIgRew',
                              check_hash=False, progress=True)
 Xclassifier.load_state_dict(torch.load(cache, map_location=torch.device('cpu')))
 Xclassifier.eval()

     classifier = classifier.cuda()
 # Load pretrained models
+cache = download_cached_file('https://czc-checkpoint.oss-cn-hangzhou.aliyuncs.com/bing/sRGB/LUTs.pth',
                              check_hash=False, progress=True)
 LUTs = torch.load(cache, map_location=torch.device('cpu'))
 LUT0.load_state_dict(LUTs["0"])
 LUT1.eval()
 LUT2.eval()
+cache = download_cached_file('https://czc-checkpoint.oss-cn-hangzhou.aliyuncs.com/bing/sRGB/classifier.pth',
                              check_hash=False, progress=True)
 classifier.load_state_dict(torch.load(cache, map_location=torch.device('cpu')))
 classifier.eval()
     Xclassifier = Xclassifier.cuda()
 # Load pretrained models
+cache = download_cached_file('https://czc-checkpoint.oss-cn-hangzhou.aliyuncs.com/bing/XYZ/LUTs.pth',
                              check_hash=False, progress=True)
 XLUTs = torch.load(cache, map_location=torch.device('cpu'))
 XLUT0.load_state_dict(XLUTs["0"])
 XLUT1.eval()
 XLUT2.eval()
+cache = download_cached_file('https://czc-checkpoint.oss-cn-hangzhou.aliyuncs.com/bing/XYZ/classifier.pth',
                              check_hash=False, progress=True)
 Xclassifier.load_state_dict(torch.load(cache, map_location=torch.device('cpu')))
 Xclassifier.eval()

models/__pycache__/models_x.cpython-310.pyc ADDED Viewed

Binary file (8.57 kB). View file

models/__pycache__/trilinear_test.cpython-310.pyc ADDED Viewed

Binary file (8.28 kB). View file