Solar_Panel_fault_V / utils /image_processing.py
Nawinkumar15's picture
Create utils/image_processing.py
be8eafa verified
import numpy as np
import cv2
from PIL import Image
def preprocess_image(image: Image.Image):
# Convert PIL to OpenCV
img = np.array(image)
if img.shape[-1] == 4:
img = img[..., :3]
return cv2.cvtColor(img, cv2.COLOR_RGB2BGR)