swz30 · dsid271 · Oct 5, 2024 · Oct 5, 2024 · Oct 5, 2024 · Oct 5, 2024
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_1.png b/datasets/dnd/dnd_rgb/noisy/0001_1.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_10.png b/datasets/dnd/dnd_rgb/noisy/0001_10.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_11.png b/datasets/dnd/dnd_rgb/noisy/0001_11.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_12.png b/datasets/dnd/dnd_rgb/noisy/0001_12.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_13.png b/datasets/dnd/dnd_rgb/noisy/0001_13.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_14.png b/datasets/dnd/dnd_rgb/noisy/0001_14.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_15.png b/datasets/dnd/dnd_rgb/noisy/0001_15.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_16.png b/datasets/dnd/dnd_rgb/noisy/0001_16.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_17.png b/datasets/dnd/dnd_rgb/noisy/0001_17.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_18.png b/datasets/dnd/dnd_rgb/noisy/0001_18.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_19.png b/datasets/dnd/dnd_rgb/noisy/0001_19.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_2.png b/datasets/dnd/dnd_rgb/noisy/0001_2.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_20.png b/datasets/dnd/dnd_rgb/noisy/0001_20.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_3.png b/datasets/dnd/dnd_rgb/noisy/0001_3.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_4.png b/datasets/dnd/dnd_rgb/noisy/0001_4.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_5.png b/datasets/dnd/dnd_rgb/noisy/0001_5.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_6.png b/datasets/dnd/dnd_rgb/noisy/0001_6.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_7.png b/datasets/dnd/dnd_rgb/noisy/0001_7.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_8.png b/datasets/dnd/dnd_rgb/noisy/0001_8.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0001_9.png b/datasets/dnd/dnd_rgb/noisy/0001_9.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_1.png b/datasets/dnd/dnd_rgb/noisy/0002_1.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_10.png b/datasets/dnd/dnd_rgb/noisy/0002_10.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_11.png b/datasets/dnd/dnd_rgb/noisy/0002_11.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_12.png b/datasets/dnd/dnd_rgb/noisy/0002_12.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_13.png b/datasets/dnd/dnd_rgb/noisy/0002_13.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_14.png b/datasets/dnd/dnd_rgb/noisy/0002_14.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_15.png b/datasets/dnd/dnd_rgb/noisy/0002_15.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_16.png b/datasets/dnd/dnd_rgb/noisy/0002_16.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_17.png b/datasets/dnd/dnd_rgb/noisy/0002_17.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_18.png b/datasets/dnd/dnd_rgb/noisy/0002_18.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_19.png b/datasets/dnd/dnd_rgb/noisy/0002_19.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_2.png b/datasets/dnd/dnd_rgb/noisy/0002_2.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_20.png b/datasets/dnd/dnd_rgb/noisy/0002_20.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_3.png b/datasets/dnd/dnd_rgb/noisy/0002_3.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_4.png b/datasets/dnd/dnd_rgb/noisy/0002_4.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_5.png b/datasets/dnd/dnd_rgb/noisy/0002_5.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_6.png b/datasets/dnd/dnd_rgb/noisy/0002_6.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_7.png b/datasets/dnd/dnd_rgb/noisy/0002_7.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_8.png b/datasets/dnd/dnd_rgb/noisy/0002_8.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0002_9.png b/datasets/dnd/dnd_rgb/noisy/0002_9.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_1.png b/datasets/dnd/dnd_rgb/noisy/0003_1.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_10.png b/datasets/dnd/dnd_rgb/noisy/0003_10.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_2.png b/datasets/dnd/dnd_rgb/noisy/0003_2.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_3.png b/datasets/dnd/dnd_rgb/noisy/0003_3.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_4.png b/datasets/dnd/dnd_rgb/noisy/0003_4.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_5.png b/datasets/dnd/dnd_rgb/noisy/0003_5.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_6.png b/datasets/dnd/dnd_rgb/noisy/0003_6.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_7.png b/datasets/dnd/dnd_rgb/noisy/0003_7.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_8.png b/datasets/dnd/dnd_rgb/noisy/0003_8.png
diff --git a/datasets/dnd/dnd_rgb/noisy/0003_9.png b/datasets/dnd/dnd_rgb/noisy/0003_9.png
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,8 @@
+cmake
+build-essential
+libjpeg-dev
+libpng-dev
+conda create -n pytorch1 python=3.7
+conda activate pytorch1
+conda install pytorch=1.1 torchvision=0.3 cudatoolkit=9.0 -c pytorch
+pip install matplotlib scikit-image yacs lycon natsort h5py tqdm
diff --git a/runtime.txt b/runtime.txt
@@ -0,0 +1 @@
+python-3.7
diff --git a/utils/image_utils.py b/utils/image_utils.py
@@ -1,9 +1,9 @@
 import torch
 import numpy as np
-from skimage.measure.simple_metrics import compare_psnr
+from skimage.metrics import peak_signal_noise_ratio
 import pickle
 import lycon
-from skimage.measure import compare_ssim
+from skimage.metrics import structural_similarity
 
 
 def is_numpy_file(filename):
@@ -47,7 +47,7 @@ def batch_PSNR(img, imclean, data_range):
     Iclean = imclean.data.cpu().numpy().astype(np.float32)
     PSNR = []
     for i in range(Img.shape[0]):
-        psnr = compare_psnr(Iclean[i,:,:,:], Img[i,:,:,:], data_range=data_range)
+        psnr =  peak_signal_noise_ratio(Iclean[i,:,:,:], Img[i,:,:,:], data_range=data_range)
         if np.isinf(psnr):
             continue
         PSNR.append(psnr)
@@ -59,7 +59,7 @@ def batch_SSIM(img, imclean):
     Iclean = imclean.data.cpu().numpy().astype(np.float32)
     SSIM = []
     for i in range(Img.shape[0]):
-        ssim = compare_ssim(Iclean[i,:,:,:], Img[i,:,:,:], gaussian_weights=True, use_sample_covariance=False, multichannel =True)
+        ssim = structural_similarity(Iclean[i,:,:,:], Img[i,:,:,:], gaussian_weights=True, use_sample_covariance=False, multichannel =True)
         SSIM.append(ssim)
     return sum(SSIM)/len(SSIM)