From e62adcf978a92ddc128245d4ccd07d098d2f82eb Mon Sep 17 00:00:00 2001 From: Stefan Klut Date: Mon, 4 Dec 2023 14:54:45 +0100 Subject: [PATCH] Incorrect saving --- eval.py | 2 +- utils/image_utils.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/eval.py b/eval.py index 1efcdae..84f1b78 100644 --- a/eval.py +++ b/eval.py @@ -36,7 +36,7 @@ def get_arguments() -> argparse.Namespace: # io_args.add_argument("-t", "--train", help="Train input folder/file", # nargs="+", action="extend", type=str, default=None) io_args.add_argument("-i", "--input", help="Input folder/file", nargs="+", action="extend", type=str, default=None) - io_args.add_argument("-o", "--output", help="Output folder", type=str, required=True) + io_args.add_argument("-o", "--output", help="Output folder", type=str) tmp_args = parser.add_argument_group("tmp files") tmp_args.add_argument("--tmp_dir", help="Temp files folder", type=str, default=None) diff --git a/utils/image_utils.py b/utils/image_utils.py index d309eb5..84dc438 100644 --- a/utils/image_utils.py +++ b/utils/image_utils.py @@ -147,12 +147,10 @@ def save_image_array_to_path( Args: image_path (Path | str): save path location - array (np.ndarray): image in array form (BGR between 0 and 255) + array (np.ndarray): image in array form (RGB between 0 and 255) """ try: # cv2.imwrite(str(image_path), array) - if len(array.shape) == 3: - array = cv2.cvtColor(array.astype(np.uint8), cv2.COLOR_BGR2RGB) image = Image.fromarray(array) image.save(image_path) except OSError: