(feat) add first step of highlighting features of interest
This commit is contained in:
24
main.py
Normal file
24
main.py
Normal 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()
|
||||
Reference in New Issue
Block a user