Astridkraft commited on
Commit
5123001
·
verified ·
1 Parent(s): 2259119

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -167,6 +167,13 @@ def create_face_mask(image, bbox_coords, mode):
167
 
168
  # === KORREKTE GEMEINSAME PROPORTIONALE SKALIERUNG MIT PADDING ===
169
  def scale_image_and_mask_together(image, mask, target_size=512):
 
 
 
 
 
 
 
170
  """
171
  SKALIERT BILD UND MASKE GEMEINSAM MIT GLEICHEN PROPORTIONEN (MIT PADDING)
172
  Behält das Seitenverhältnis bei und fügt ggf. Padding hinzu
@@ -181,8 +188,6 @@ def scale_image_and_mask_together(image, mask, target_size=512):
181
  - padded_mask: skalierte Maske mit Padding (L)
182
  - padding_info: Dictionary mit Skalierungsinfo für späteres Compositing
183
  """
184
- if image is None or mask is None:
185
- raise ValueError("Bild oder Maske ist None")
186
 
187
  if image.size != mask.size:
188
  raise ValueError(f"Bild und Maske haben unterschiedliche Größen: {image.size} vs {mask.size}") #Stoppt Programm sofort mit Fehlermeldung!
 
167
 
168
  # === KORREKTE GEMEINSAME PROPORTIONALE SKALIERUNG MIT PADDING ===
169
  def scale_image_and_mask_together(image, mask, target_size=512):
170
+ if image is None or mask is None:
171
+ raise ValueError("Bild oder Maske ist None")
172
+
173
+ # MAX SIZE CHECK HINZUFÜGEN
174
+ if image.width > MAX_IMAGE_SIZE or image.height > MAX_IMAGE_SIZE:
175
+ raise ValueError(f"Bild zu groß! Maximal {MAX_IMAGE_SIZE}×{MAX_IMAGE_SIZE}")
176
+
177
  """
178
  SKALIERT BILD UND MASKE GEMEINSAM MIT GLEICHEN PROPORTIONEN (MIT PADDING)
179
  Behält das Seitenverhältnis bei und fügt ggf. Padding hinzu
 
188
  - padded_mask: skalierte Maske mit Padding (L)
189
  - padding_info: Dictionary mit Skalierungsinfo für späteres Compositing
190
  """
 
 
191
 
192
  if image.size != mask.size:
193
  raise ValueError(f"Bild und Maske haben unterschiedliche Größen: {image.size} vs {mask.size}") #Stoppt Programm sofort mit Fehlermeldung!