qudida_domain_adapter_color_distribution_transfer_with_pca.py

1import cv2
2from qudida import DomainAdapter
3
4# Load the source image (the image you want to transform)
5# and the target image (the image whose color distribution you want to copy)
6source_image = cv2.imread("source.jpg")
7target_image = cv2.imread("target.jpg")
8
9# Convert images to RGB if they were loaded in BGR by OpenCV
10source_image = cv2.cvtColor(source_image, cv2.COLOR_BGR2RGB)
11target_image = cv2.cvtColor(target_image, cv2.COLOR_BGR2RGB)
12
13# Initialize the DomainAdapter
14adapter = DomainAdapter(transformer_type="pca")
15
16# Fit the adapter to the target image and transform the source image
17# This matches the source image's statistics to the target image
18transformed_image = adapter(source_image, target_image)
19
20# Save or display the result
21# transformed_image is a numpy array with the same shape as the source_image
22transformed_image_bgr = cv2.cvtColor(transformed_image, cv2.COLOR_RGB2BGR)
23cv2.imwrite("result.jpg", transformed_image_bgr)