ניסיון תיקון

backend.py

@@ -7,8 +7,6 @@ import numpy as np
 import cv2
 from PIL import Image, ImageFilter
 from scipy.ndimage import binary_dilation
-import hydra
-from hydra import initialize_config_dir, compose
 from omegaconf import OmegaConf
 
 # -----------------------------
@@ -158,23 +156,28 @@ CONF_THRESHOLD = 0.2
 # -----------------------------
 # 4) הכנה ל-SAM2
 # -----------------------------
-try:
-    from hydra import initialize
-    from sam2.sam2_image_predictor import SAM2ImagePredictor
+from sam2.build_sam import build_sam2
+from sam2.sam2_image_predictor import SAM2ImagePredictor
 
-    SAM2_CONFIG_PATH = "sam2.1/"
-    SAM2_MODEL_NAME = "facebook/sam2.1-hiera-tiny"
+# נתיבים יחסיים ל-Space של Hugging Face
+SAM2_CHECKPOINT = "checkpoints/sam2.1_hiera_tiny.pt"  # עדכן את הנתיב בהתאם
+MODEL_CFG = "configs/sam2.1_hiera_t.yaml"
 
-    sam2_predictor = None
-    device = "cuda" if torch.cuda.is_available() else "cpu"
+sam2_predictor = None
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
-    with initialize(config_path=SAM2_CONFIG_PATH):
-        sam2_predictor = SAM2ImagePredictor.from_pretrained(SAM2_MODEL_NAME)
-    sam2_predictor.model.to(device)
+try:
+    sam2_model = build_sam2(MODEL_CFG, SAM2_CHECKPOINT, device=device)
+    sam2_predictor = SAM2ImagePredictor(sam2_model)
+    print("[SAM2] מודל SAM2 נטען בהצלחה.")
 
 except Exception as e:
-    print("[SAM2] לא מצליח לטעון את SAM2. ודא שהנתיב והקונפיג נכונים.")
-    raise  # העברת השגיאה הלאה
+    print(f"[SAM2] שגיאה בטעינת SAM2: {e}")
+    print(f"  - סוג השגיאה: {type(e).__name__}")
+    print(f"  - הודעת השגיאה: {e}")
+    import traceback
+    print(f"  - Traceback:")
+    traceback.print_exc()
     sam2_predictor = None
 
 # -----------------------------
@@ -231,13 +234,7 @@ def process_image(
     if not gemini_api_key:
         raise ValueError("מפתח API של Gemini אינו מוזן.")
     """
-    פונקציה המקבלת תמונת PIL, מפתח API של Gemini, ומחזירה את התמונה לאחר טשטוש נשים,
-    תוך שלבי התקדמות מוגדרים:
-      - זיהוי אנשים ב-YOLO
-      - זיהוי אם אישה בעזרת Gemini
-      - פילוח באמצעות SAM2
-      - טשטוש
-    פרמטר progress_callback: פונקציה לקבלת (fraction, description)
+    פונקציה המקבלת תמונת PIL, מפתח API של Gemini, ומחזירה את התמונה לאחר טשטוש נשים.
     """
 
     if progress_callback is None:
@@ -287,24 +284,7 @@ def process_image(
             women_boxes.append(bbox)
 
     # 3) שלב SAM2 (עבור בוקסים של נשים)
-    # אתחול Hydra בתוך הפונקציה
-    global sam2_predictor
-    if sam2_predictor is None:
-        try:
-            config_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), SAM2_CONFIG_PATH))
-            with initialize_config_dir(version_base=None, config_dir=config_dir):
-                cfg = compose(config_name="config") # "config" is the name of the config file, replace if necessary
-                sam2_predictor = SAM2ImagePredictor.from_pretrained(SAM2_MODEL_NAME, config=cfg)
-                sam2_predictor.model.to(device)
-        except Exception as e:
-            print(f"[SAM2] שגיאה בטעינת SAM2: {e}")
-            print(f"  - סוג השגיאה: {type(e).__name__}")
-            print(f"  - הודעת השגיאה: {e}")
-            import traceback
-            print(f"  - Traceback:")
-            traceback.print_exc()
-            sam2_predictor = None
-
+    # אין צורך לאתחל Hydra, פשוט משתמשים ב-sam2_predictor
     if sam2_predictor is None:
         print("[process_image] SAM2 לא זמין/נטען. מחזירים תמונה ללא טשטוש.")
         raise ValueError("SAM2 model is not loaded.")

checkpoints/sam2.1_hiera_tiny.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7402e0d864fa82708a20fbd15bc84245c2f26dff0eb43a4b5b93452deb34be69
+size 156008466

instruct.txt

@@ -0,0 +1,28 @@
+C:.
+│   .gitattributes
+│   .gitignore
+│   app.py
+│   backend.py
+│   README.md
+│   requirements.txt
+│
+├───checkpoints
+├───configs
+│       sam2.1_hiera_t.yaml
+│
+├───example_images
+│       example.jpg
+│
+├───sam2.1
+│   ├───checkpoints
+│   └───configs
+│       └───sam2.1
+│               sam2.1_hiera_b+.yaml
+│               sam2.1_hiera_l.yaml
+│               sam2.1_hiera_s.yaml
+│               sam2.1_hiera_t.yaml
+│
+└───__pycache__
+        app.cpython-310.pyc
+        backend.cpython-310.pyc
+        webui.cpython-310.pyc

requirements.txt

@@ -6,8 +6,7 @@ Pillow
 requests
 ultralytics
 scipy
-hydra-core
+omegaconf
 torchvision
 supervision
-git+https://github.com/facebookresearch/sam2.git
-omegaconf
+git+https://github.com/facebookresearch/sam2.git