Merge pull request #76 from Tsubajashi/main

comment skimage entries
This commit is contained in:
N00MKRAD
2021-08-24 23:45:55 +02:00
committed by GitHub

View File

@@ -7,8 +7,8 @@ import torch.nn as nn
import torch.nn.functional as F
from datetime import datetime
from torch.nn import init
from skimage.measure import compare_ssim
from skimage.metrics import structural_similarity
#from skimage.measure import compare_ssim
#from skimage.metrics import structural_similarity
from torch.autograd import Variable
from torchvision import models