Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
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")
|