Update app.py
Browse files
app.py
CHANGED
|
@@ -11,6 +11,7 @@ import os
|
|
| 11 |
import tempfile
|
| 12 |
import random
|
| 13 |
import re
|
|
|
|
| 14 |
|
| 15 |
# === OPTIMIERTE EINSTELLUNGEN ===
|
| 16 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
@@ -178,8 +179,9 @@ def create_face_mask(image, bbox_coords, mode):
|
|
| 178 |
Returns:
|
| 179 |
- padded_image: skaliertes Bild mit Padding (RGB)
|
| 180 |
- padded_mask: skalierte Maske mit Padding (L)
|
| 181 |
-
- padding_info: Dictionary mit Skalierungsinfo für späteres Compositing
|
| 182 |
"""
|
|
|
|
| 183 |
def scale_image_and_mask_together(image, mask, target_size=512):
|
| 184 |
if image is None or mask is None:
|
| 185 |
raise ValueError("Bild oder Maske ist None")
|
|
@@ -237,7 +239,6 @@ def scale_image_and_mask_together(image, mask, target_size=512):
|
|
| 237 |
|
| 238 |
|
| 239 |
|
| 240 |
-
|
| 241 |
# Composition Workflow nach Ausgabe ControlnetInpaint-Pipeline
|
| 242 |
def enhanced_composite_with_sam(original_image, inpaint_result, original_mask,
|
| 243 |
padding_info, bbox_coords, mode):
|
|
|
|
| 11 |
import tempfile
|
| 12 |
import random
|
| 13 |
import re
|
| 14 |
+
from PIL import ImageFilter # Für GaussianBlur wird nur für SAM benötigt!
|
| 15 |
|
| 16 |
# === OPTIMIERTE EINSTELLUNGEN ===
|
| 17 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 179 |
Returns:
|
| 180 |
- padded_image: skaliertes Bild mit Padding (RGB)
|
| 181 |
- padded_mask: skalierte Maske mit Padding (L)
|
| 182 |
+
- padding_info: Dictionary mit Skalierungsinfo für späteres Compositing
|
| 183 |
"""
|
| 184 |
+
|
| 185 |
def scale_image_and_mask_together(image, mask, target_size=512):
|
| 186 |
if image is None or mask is None:
|
| 187 |
raise ValueError("Bild oder Maske ist None")
|
|
|
|
| 239 |
|
| 240 |
|
| 241 |
|
|
|
|
| 242 |
# Composition Workflow nach Ausgabe ControlnetInpaint-Pipeline
|
| 243 |
def enhanced_composite_with_sam(original_image, inpaint_result, original_mask,
|
| 244 |
padding_info, bbox_coords, mode):
|