16 lines
883 B
Python
16 lines
883 B
Python
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) |