utils/image_io.py

import cv2
import uuid
import os


OUTPUT_DIR = "outputs"
os.makedirs(OUTPUT_DIR, exist_ok=True)


def save_bmp(image):

    filename = os.path.join(
        OUTPUT_DIR,
        f"{uuid.uuid4()}.bmp"
    )

    cv2.imwrite(
        filename,
        image,
        [cv2.IMWRITE_BMP_COMPRESSION, 0]
    )

    return filename


def save_png(image):

    filename = os.path.join(
        OUTPUT_DIR,
        f"{uuid.uuid4()}.png"
    )

    cv2.imwrite(
        filename,
        image
    )

    return filename