mlbench123 commited on
Commit
282e464
·
verified ·
1 Parent(s): 7942e96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -965,7 +965,7 @@ def predict_with_paper(image, paper_size, offset, offset_unit, finger_clearance=
965
 
966
  try:
967
  # Mask paper area in input image first
968
- masked_input_image = mask_paper_area_in_image(image, paper_contour)
969
 
970
  # Use YOLOv8 to detect objects
971
  yolo_v8 = get_yolo_v8()
@@ -975,7 +975,7 @@ def predict_with_paper(image, paper_size, offset, offset_unit, finger_clearance=
975
  crop_offset = (0, 0)
976
  else:
977
  # YOLOv8 detects all COCO classes by default
978
- results = yolo_v8.predict(masked_input_image, conf=0.1, verbose=False)
979
 
980
  if not results or len(results) == 0 or not hasattr(results[0], 'boxes') or len(results[0].boxes) == 0:
981
  logger.warning("No objects detected by YOLOv8, proceeding with full image")
 
965
 
966
  try:
967
  # Mask paper area in input image first
968
+ # masked_input_image = mask_paper_area_in_image(image, paper_contour)
969
 
970
  # Use YOLOv8 to detect objects
971
  yolo_v8 = get_yolo_v8()
 
975
  crop_offset = (0, 0)
976
  else:
977
  # YOLOv8 detects all COCO classes by default
978
+ results = yolo_v8.predict(image, conf=0.1, verbose=False)
979
 
980
  if not results or len(results) == 0 or not hasattr(results[0], 'boxes') or len(results[0].boxes) == 0:
981
  logger.warning("No objects detected by YOLOv8, proceeding with full image")