fixes here and there

config/config.yaml

@@ -3,7 +3,8 @@ project: deblur_gan
 experiment_desc: fpn
 
 train:
-  files_a: &FILES_A /datasets/my_dataset/**/*.jpg
+#  files_a: &FILES_A /datasets/my_dataset/**/*.jpg
+  files_a: &FILES_A /home/arseny/kp_data/train2017/*.jpg
   files_b: *FILES_A
   size: &SIZE 256
   crop: random
@@ -12,20 +13,13 @@ train:
   bounds: [0, .9]
   scope: geometric
   corrupt: &CORRUPT
-    - name: cutout
-      prob: 0.5
-      num_holes: 3
-      max_h_size: 25
-      max_w_size: 25
     - name: jpeg
-      quality_lower: 70
+      quality_lower: 50
       quality_upper: 90
     - name: motion_blur
     - name: median_blur
-    - name: gamma
-    - name: rgb_shift
-    - name: hsv_shift
     - name: sharpen
+    - name: pixelize
 
 val:
   files_a: *FILES_A
@@ -41,7 +35,7 @@ val:
 phase: train
 warmup_num: 3
 model:
-  g_name: fpn_inception
+  g_name: fpn_dense
   blocks: 9
   d_name: double_gan # may be no_gan, patch_gan, double_gan, multi_scale
   d_layers: 3

models/models.py

@@ -1,7 +1,6 @@
 import numpy as np
 import torch.nn as nn
-from skimage.measure import compare_ssim as SSIM
-
+from skimage.metrics import structural_similarity as SSIM
 from util.metrics import PSNR