Asset Analysis Complete + Generation Plan

scripts/remove_bg_simple.py

@@ -0,0 +1,43 @@
+
+import cv2
+import numpy as np
+import os
+
+FILES_TO_PROCESS = [
+    "/Users/davidkotnik/repos/novafarma/nova farma TRAE/assets/DEMO_FAZA1/Environment/water_clean_patch.png"
+]
+
+def remove_white_bg(file_path):
+    print(f"Processing: {file_path}")
+    if not os.path.exists(file_path):
+        print(f"File not found: {file_path}")
+        return
+
+    # Read image
+    img = cv2.imread(file_path, cv2.IMREAD_COLOR)
+    if img is None:
+        print("Failed to load image.")
+        return
+
+    # Convert to Gray
+    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+
+    # Create mask: White (240-255) becomes transparent (0), everything else opaque (255)
+    # THRESH_BINARY_INV: Values > 240 become 0, others 255.
+    _, mask = cv2.threshold(gray, 240, 255, cv2.THRESH_BINARY_INV)
+
+    # refine mask to remove jagged edges if possible (optional)
+    # kernel = np.ones((2,2), np.uint8)
+    # mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel) 
+
+    # Add Alpha Channel
+    b, g, r = cv2.split(img)
+    rgba = cv2.merge([b, g, r, mask])
+
+    # Save back to same path
+    cv2.imwrite(file_path, rgba)
+    print("Saved with transparency.")
+
+if __name__ == "__main__":
+    for f in FILES_TO_PROCESS:
+        remove_white_bg(f)

scripts/remove_magenta_specific.py

@@ -0,0 +1,51 @@
+
+import cv2
+import numpy as np
+
+FILE_PATH = "/Users/davidkotnik/repos/novafarma/nova farma TRAE/assets/DEMO_FAZA1/Environment/water_clean_patch.png"
+
+def remove_magenta_bg(file_path):
+    print(f"Processing {file_path}")
+    img = cv2.imread(file_path, cv2.IMREAD_UNCHANGED)
+    if img is None:
+        print("Failed to load image")
+        return
+
+    # If already RGBA, keep alpha. If RGB, add alpha.
+    if img.shape[2] == 3:
+        b, g, r = cv2.split(img)
+        a = np.ones_like(b) * 255
+        img = cv2.merge((b, g, r, a))
+    
+    # Convert to HSV
+    hsv = cv2.cvtColor(img[:,:,0:3], cv2.COLOR_BGR2HSV)
+    
+    # Magenta #FF00FF is (300 deg, 100%, 100%). In OpenCV (0-179, 0-255, 0-255):
+    # Hue ~ 150.
+    # Define range for Magenta
+    lower_magenta = np.array([145, 150, 150])
+    upper_magenta = np.array([165, 255, 255])
+    
+    mask = cv2.inRange(hsv, lower_magenta, upper_magenta)
+    
+    # Invert mask (keep non-magenta)
+    mask_inv = cv2.bitwise_not(mask)
+    
+    # Get current alpha
+    b, g, r, a = cv2.split(img)
+    
+    # Combine existing alpha with new mask
+    new_alpha = cv2.bitwise_and(a, mask_inv)
+    
+    # Soften edges (1px blur on alpha)
+    # This helps remove jagged pink pixels
+    new_alpha = cv2.GaussianBlur(new_alpha, (3, 3), 0)
+    
+    # Update image
+    final_img = cv2.merge((b, g, r, new_alpha))
+    
+    cv2.imwrite(file_path, final_img)
+    print("Background removed and saved.")
+
+if __name__ == "__main__":
+    remove_magenta_bg(FILE_PATH)