Astridkraft commited on
Commit
0b3661f
·
verified ·
1 Parent(s): 336f598

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +8 -10
controlnet_module.py CHANGED
@@ -646,7 +646,7 @@ class ControlNetProcessor:
646
  # ============================================================
647
  original_image = image
648
  print(f"💾 Originalbild gesichert: {original_image.size}")
649
- original_bbox = (x1, y1, x2, y2) # <-- DAS FEHLT
650
  print(f"💾 Original-BBox gespeichert: {original_bbox}")
651
 
652
 
@@ -664,7 +664,7 @@ class ControlNetProcessor:
664
 
665
 
666
  if use_crop_strategy:
667
- #
668
  # ============================================================
669
  # Crop = BBox × 2.5 (ERHÖHT für mehr Kontext)
670
  # ============================================================
@@ -817,7 +817,7 @@ class ControlNetProcessor:
817
  print(f" • Schärfe: +100%")
818
  print(f" • Helligkeit: +10%")
819
 
820
- # Für SAM: Verwende aufbereiteten Ausschnitt
821
  image = enhanced_image
822
  x1, y1, x2, y2 = rel_x1, rel_y1, rel_x2, rel_y2
823
 
@@ -839,7 +839,8 @@ class ControlNetProcessor:
839
 
840
 
841
  # ============================================================
842
- # SAM-AUSFÜHRUNG mit relativen Koordinaten für Bild<=512 und Bild>512
 
843
  # ============================================================
844
 
845
  print("-" * 60)
@@ -934,9 +935,9 @@ class ControlNetProcessor:
934
 
935
 
936
  # ============================================================
937
- # HEURISTIK (zur Bestimmung der besten Maske)
938
  # ============================================================
939
- print("🤔 HEURISTIK AUF Original- GRÖSSE BERECHNEN")
940
 
941
  if use_crop_strategy:
942
  # BBox-Information für Heuristik (IN CROP-KOORDINATEN!)
@@ -984,7 +985,7 @@ class ControlNetProcessor:
984
  print(f" ❌ Maske {i+1}: Keine Pixel nach Threshold {adaptive_threshold:.3f}")
985
  continue
986
 
987
- #Maskenfläche in Pixeln (Crop-Grösse)
988
  mask_area_pixels = np.sum(mask_binary)
989
 
990
  # ============================================================
@@ -1099,9 +1100,6 @@ class ControlNetProcessor:
1099
  mask_np = all_masks_crop[best_mask_idx]
1100
  max_val = mask_np.max()
1101
  print(f"🔍 Maximaler SAM-Konfidenzwert der besten Maske: {max_val:.3f}")
1102
-
1103
-
1104
-
1105
 
1106
 
1107
  # ============================================================
 
646
  # ============================================================
647
  original_image = image
648
  print(f"💾 Originalbild gesichert: {original_image.size}")
649
+ original_bbox = (x1, y1, x2, y2)
650
  print(f"💾 Original-BBox gespeichert: {original_bbox}")
651
 
652
 
 
664
 
665
 
666
  if use_crop_strategy:
667
+
668
  # ============================================================
669
  # Crop = BBox × 2.5 (ERHÖHT für mehr Kontext)
670
  # ============================================================
 
817
  print(f" • Schärfe: +100%")
818
  print(f" • Helligkeit: +10%")
819
 
820
+ # Für SAM: Verwende aufbereitetes Crop-Bild und transformierte BBox-Koordinaten
821
  image = enhanced_image
822
  x1, y1, x2, y2 = rel_x1, rel_y1, rel_x2, rel_y2
823
 
 
839
 
840
 
841
  # ============================================================
842
+ # SAM-AUSFÜHRUNG mit transformierter BBox und aufbereiteten Crop-Bild für Bild >512 oder
843
+ # SAM-AUSFÜHRUNG mit Originalbild und BBox für Bilder <=512
844
  # ============================================================
845
 
846
  print("-" * 60)
 
935
 
936
 
937
  # ============================================================
938
+ # HEURISTIK (zur Bestimmung der besten Maske) bei Bild>512 auf Crop-Größe bei Bild<=512 auf Originalgröße
939
  # ============================================================
940
+ print("🤔 HEURISTIK BERECHNEN")
941
 
942
  if use_crop_strategy:
943
  # BBox-Information für Heuristik (IN CROP-KOORDINATEN!)
 
985
  print(f" ❌ Maske {i+1}: Keine Pixel nach Threshold {adaptive_threshold:.3f}")
986
  continue
987
 
988
+ #Maskenfläche in Pixeln
989
  mask_area_pixels = np.sum(mask_binary)
990
 
991
  # ============================================================
 
1100
  mask_np = all_masks_crop[best_mask_idx]
1101
  max_val = mask_np.max()
1102
  print(f"🔍 Maximaler SAM-Konfidenzwert der besten Maske: {max_val:.3f}")
 
 
 
1103
 
1104
 
1105
  # ============================================================