Update controlnet_module.py
Browse files- 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)
|
| 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
|
| 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
|
|
|
|
| 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
|
| 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
|
| 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 |
# ============================================================
|