Spaces:
Sleeping
Sleeping
File size: 371 Bytes
dd85fb6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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)
|