diff options
Diffstat (limited to 'test-mogrify.py')
| -rw-r--r-- | test-mogrify.py | 13 |
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) |