import cv2 import numpy as np def to_bgr(img: np.ndarray) -> np.ndarray: if img.ndim == 2: return cv2.cvtColor(img, cv2.COLOR_GRAY2BGR) if img.shape[2] == 4: return cv2.cvtColor(img, cv2.COLOR_RGBA2BGR) return cv2.cvtColor(img, cv2.COLOR_RGB2BGR) def to_rgb(img: np.ndarray) -> np.ndarray: return cv2.cvtColor(img, cv2.COLOR_BGR2RGB)