Astridkraft commited on
Commit
a40fa2d
·
verified ·
1 Parent(s): 8153302

Update sam_module.py

Browse files
Files changed (1) hide show
  1. sam_module.py +3 -0
sam_module.py CHANGED
@@ -316,14 +316,17 @@ def create_sam_mask(self, image, bbox_coords, mode):
316
  return self._create_rectangular_mask(image, bbox_coords, "focus_change")
317
 
318
  except Exception as e:
 
319
  print("❌ FEHLER IN SAM 2 SEGMENTIERUNG")
320
  print(f"Fehler: {str(e)[:200]}")
 
321
  import traceback
322
  traceback.print_exc()
323
 
324
  # Fallback
325
  fallback_mask = self._create_rectangular_mask(original_image, original_bbox, mode)
326
  if fallback_mask.size != original_image.size:
 
327
  fallback_mask = fallback_mask.resize(original_image.size, Image.Resampling.NEAREST)
328
 
329
  return fallback_mask, fallback_mask
 
316
  return self._create_rectangular_mask(image, bbox_coords, "focus_change")
317
 
318
  except Exception as e:
319
+ print("❌" * 40)
320
  print("❌ FEHLER IN SAM 2 SEGMENTIERUNG")
321
  print(f"Fehler: {str(e)[:200]}")
322
+ print("❌" * 40)
323
  import traceback
324
  traceback.print_exc()
325
 
326
  # Fallback
327
  fallback_mask = self._create_rectangular_mask(original_image, original_bbox, mode)
328
  if fallback_mask.size != original_image.size:
329
+ print(f" ⚠️ Fallback-Maske angepasst: {fallback_mask.size} → {original_image.size}")
330
  fallback_mask = fallback_mask.resize(original_image.size, Image.Resampling.NEAREST)
331
 
332
  return fallback_mask, fallback_mask