Update model/pytorch_msssim/__init__.py
Browse files
model/pytorch_msssim/__init__.py
CHANGED
|
@@ -2,8 +2,9 @@ import torch
|
|
| 2 |
import torch.nn.functional as F
|
| 3 |
from math import exp
|
| 4 |
import numpy as np
|
| 5 |
-
|
| 6 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 7 |
|
| 8 |
def gaussian(window_size, sigma):
|
| 9 |
gauss = torch.Tensor([exp(-(x - window_size//2)**2/float(2*sigma**2)) for x in range(window_size)])
|
|
|
|
| 2 |
import torch.nn.functional as F
|
| 3 |
from math import exp
|
| 4 |
import numpy as np
|
| 5 |
+
import super_image
|
| 6 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 7 |
+
@spaces.GPU(duration=5)
|
| 8 |
|
| 9 |
def gaussian(window_size, sigma):
|
| 10 |
gauss = torch.Tensor([exp(-(x - window_size//2)**2/float(2*sigma**2)) for x in range(window_size)])
|