25 lines
786 B
Python
25 lines
786 B
Python
import cv2
|
|
|
|
from src.data import load_data
|
|
from src.img_utils import apply_color_overlay, blur_background, desaturate_background
|
|
|
|
|
|
def main():
|
|
print("Starting Abduction Demo")
|
|
labeled_images = load_data()
|
|
image = labeled_images[0].image
|
|
mask = labeled_images[0].create_mask([labeled_images[0].labels[0]])
|
|
|
|
overlay_result = apply_color_overlay(image, mask, color=(255, 100, 0), alpha=0.4)
|
|
cv2.imwrite(".tmp-data/highlight_overlay.jpg", overlay_result)
|
|
|
|
blur_result = blur_background(image, mask, blur_intensity=(51, 51))
|
|
cv2.imwrite(".tmp-data/highlight_blurred.jpg", blur_result)
|
|
|
|
desaturate_result = desaturate_background(image, mask)
|
|
cv2.imwrite(".tmp-data/highlight_desaturated.jpg", desaturate_result)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|