testy

1 files changed, 8 insertions, 8 deletions

diff --git a/test-mogrify.py b/test-mogrify.py
index d4cc2a6..25a62d4 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -70,6 +70,8 @@ if __name__ == '__main__':
         print("Sequence not found")
 
   def process_image(im):
+    src_img = im[:, :, ::-1].copy()
+
     img = im[:, :, ::-1].copy()
 
     if data_opt.clahe is True:
@@ -79,7 +81,6 @@ if __name__ == '__main__':
       l = clahe.apply(l)
       limg = cv2.merge((l,a,b))
       img = cv2.cvtColor(limg, cv2.COLOR_LAB2BGR)
-
     if data_opt.posterize is True:
       img = cv2.pyrMeanShiftFiltering(img, data_opt.spatial_window, data_opt.color_window)
     if data_opt.grayscale is True:
@@ -89,6 +90,12 @@ if __name__ == '__main__':
     if data_opt.canny is True:
       img = cv2.Canny(img, data_opt.canny_lo, data_opt.canny_hi)
 
+    frac_a = data_opt.process_frac
+    frac_b = 1.0 - frac_a
+    array_a = np.multiply(src_img.astype('float64'), frac_a)
+    array_b = np.multiply(img.astype('float64'), frac_b)
+    img = np.add(array_a, array_b).astype('uint8')
+
     return img
 
   if data_opt.sequence:
@@ -176,15 +183,8 @@ if __name__ == '__main__':
       # im = np.asarray(image_pil).astype('uint8')
       #print(im.shape, im.dtype)
 
-      src_img = im[:, :, ::-1].copy()
       img = process_image(im)
 
-      frac_a = data_opt.process_frac
-      frac_b = 1.0 - frac_a
-      array_a = np.multiply(src_img.astype('float64'), frac_a)
-      array_b = np.multiply(img.astype('float64'), frac_b)
-      img = np.add(array_a, array_b).astype('uint8')
-
       cv2.imwrite(tmp_path, img)
       os.rename(tmp_path, edges_path)