add custom blur kernels selection

src/rstor/analyzis/interactive/degradation.py

@@ -50,7 +50,7 @@ def get_blur_kernel_box(ksize=3):
 
 
 @interactive(
-    blur_index=(-1, [-1, 1000])
+    blur_index=(0, [-1, 2])
 )
 def get_blur_kernel(blur_index: int = -1, global_params={}):
     if blur_index == -1:
@@ -59,7 +59,7 @@ def get_blur_kernel(blur_index: int = -1, global_params={}):
     if blur_mat is False:
         blur_mat = loadmat(DATASET_BLUR_KERNEL_PATH)["kernels"].squeeze()
         global_params["blur_mat"] = blur_mat
-    blur_k = blur_mat[blur_index]
+    blur_k = blur_mat[blur_index % len(blur_mat)]
     blur_k = blur_k/blur_k.sum()
     return blur_k
 

src/rstor/properties.py

@@ -59,7 +59,7 @@ DATASET_DL_DIV2K_512 = "deadleaves_div2k_512"
 DATASET_DL_EXTRAPRIMITIVES_DIV2K_512 = "deadleaves_primitives_div2k_512"
 DATASET_SYNTH_LIST = [DATASET_DL_DIV2K_512, DATASET_DL_DIV2K_1024,
                       DATASET_DL_RANDOMRGB_1024, DATASET_DL_EXTRAPRIMITIVES_DIV2K_512]
-DATASET_BLUR_KERNEL_PATH = DATASET_PATH / "kernels" / "custom_blur_centered.mat"
+DATASET_BLUR_KERNEL_PATH = DATASET_PATH / "kernels" / "custom_blur_centered_selection.mat"
 AUGMENTATION_FLIP = "flip"
 AUGMENTATION_ROTATE = "rotate"