testy

1 files changed, 11 insertions, 2 deletions

diff --git a/test-mogrify.py b/test-mogrify.py
index 25a62d4..6b61f62 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -70,11 +70,11 @@ if __name__ == '__main__':
         print("Sequence not found")
 
   def process_image(im):
-    src_img = im[:, :, ::-1].copy()
-
     img = im[:, :, ::-1].copy()
+    processed = False
 
     if data_opt.clahe is True:
+      processed = True
       lab = cv2.cvtColor(img, cv2.COLOR_BGR2LAB)
       l, a, b = cv2.split(lab)
       clahe = cv2.createCLAHE(clipLimit=data_opt.clip_limit, tileGridSize=(8,8))
@@ -82,14 +82,23 @@ if __name__ == '__main__':
       limg = cv2.merge((l,a,b))
       img = cv2.cvtColor(limg, cv2.COLOR_LAB2BGR)
     if data_opt.posterize is True:
+      processed = True
       img = cv2.pyrMeanShiftFiltering(img, data_opt.spatial_window, data_opt.color_window)
     if data_opt.grayscale is True:
+      processed = True
       img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
     if data_opt.blur is True:
+      processed = True
       img = cv2.GaussianBlur(img, (data_opt.blur_radius, data_opt.blur_radius), data_opt.blur_sigma)
     if data_opt.canny is True:
+      processed = True
       img = cv2.Canny(img, data_opt.canny_lo, data_opt.canny_hi)
 
+    if processed is False:
+      print("No processing occurred.")
+      return img
+
+    src_img = im[:, :, ::-1].copy()
     frac_a = data_opt.process_frac
     frac_b = 1.0 - frac_a
     array_a = np.multiply(src_img.astype('float64'), frac_a)