Spaces:
Runtime error
Runtime error
from skimage.exposure import match_histograms | |
import cv2 | |
def maintain_colors(prev_img, color_match_sample, mode): | |
if mode == 'Match Frame 0 RGB': | |
return match_histograms(prev_img, color_match_sample, multichannel=True) | |
elif mode == 'Match Frame 0 HSV': | |
prev_img_hsv = cv2.cvtColor(prev_img, cv2.COLOR_RGB2HSV) | |
color_match_hsv = cv2.cvtColor(color_match_sample, cv2.COLOR_RGB2HSV) | |
matched_hsv = match_histograms(prev_img_hsv, color_match_hsv, multichannel=True) | |
return cv2.cvtColor(matched_hsv, cv2.COLOR_HSV2RGB) | |
else: # Match Frame 0 LAB | |
prev_img_lab = cv2.cvtColor(prev_img, cv2.COLOR_RGB2LAB) | |
color_match_lab = cv2.cvtColor(color_match_sample, cv2.COLOR_RGB2LAB) | |
matched_lab = match_histograms(prev_img_lab, color_match_lab, multichannel=True) | |
return cv2.cvtColor(matched_lab, cv2.COLOR_LAB2RGB) |