From 43a345987bfca3382f356a5eb8c0eec19f41fe06 Mon Sep 17 00:00:00 2001 From: Bryant Date: Fri, 5 May 2023 11:45:45 -0700 Subject: [PATCH] Resolved issues generating images and blurred image option. --- main.py | 10 ++++++++-- ops/blur.py | 4 ++-- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index f50954b..060bdba 100644 --- a/main.py +++ b/main.py @@ -10,6 +10,8 @@ from ops.noise import Noise from ops.translate import Translate from skimage.io import imread, imsave +import numpy as np +from skimage.exposure import rescale_intensity EXTENSIONS = ['png', 'jpg', 'jpeg', 'bmp'] WORKER_COUNT = max(os.cpu_count() - 1, 1) @@ -36,14 +38,18 @@ def build_augmented_file_name(original_name, ops): def work(d, f, op_lists): try: - in_path = os.path.join(d,f) + in_path = os.path.join(d, f) for op_list in op_lists: out_file_name = build_augmented_file_name(f, op_list) - if isfile(os.path.join(d,out_file_name)): + if isfile(os.path.join(d, out_file_name)): continue img = imread(in_path) for op in op_list: img = op.process(img) + + # Convert image data type to uint8 before saving + img = rescale_intensity(img, out_range=(0, 255)) + img = img.astype(np.uint8) imsave(os.path.join(d, out_file_name), img) counter.processed() diff --git a/ops/blur.py b/ops/blur.py index d6469ff..67bd9bb 100644 --- a/ops/blur.py +++ b/ops/blur.py @@ -11,8 +11,8 @@ def __init__(self, sigma): self.sigma = sigma def process(self, img): - is_colour = len(img.shape)==3 - return rescale_intensity(gaussian(img, sigma=self.sigma, multichannel=is_colour)) + is_colour = len(img.shape) == 3 + return gaussian(img, sigma=self.sigma, channel_axis=-1 if is_colour else None) @staticmethod def match_code(code):