Spaces:
Running
Running
File size: 572 Bytes
40aaca9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import cv2
import numpy as np
def preprocess_images(images, V_FIXED = 200):
fixed_images = []
for image in images:
hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
hsv_fixed = hsv.copy()
hsv_fixed[:, :, 2] = (hsv[:, :, 2] / hsv[:, :, 2].max()) * V_FIXED
hsv_fixed[:, :, 1] = hsv_fixed[:, :, 1] * (hsv_fixed[:, :, 2] / hsv[:, :, 2].max())
hsv_fixed[:, :, 1] = np.clip(hsv_fixed[:, :, 1], 0, 255)
fixed_image = cv2.cvtColor(hsv_fixed, cv2.COLOR_HSV2BGR)
fixed_images.append(fixed_image)
return fixed_images
|