(feat) add first step of highlighting features of interest

This commit is contained in:
Nicolas Sebastian Schuler
2025-07-31 12:18:18 +02:00
parent c6e30c6be6
commit 64385c1864
12 changed files with 406 additions and 64 deletions

24
main.py Normal file
View File

@@ -0,0 +1,24 @@
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()