Update handlers/frame_handler_yolo.py

handlers/frame_handler_yolo.py

@@ -68,7 +68,7 @@ def extract_key_frames(input_folder, key_frames_folder, original_fps, model_path
         if counter % 1000 == 0:
             print(f"Processed {counter} frames.")
         # Run YOLO inference
-        results = model.predict(frame, conf=0.3, verbose=False)
+        results = model.predict(frame, conf=0.1, verbose=False)
 
         # Check if a football (sports ball) is detected
         ball_detected = any(model.names[int(box.cls)] == "sports ball" for box in results[0].boxes)
@@ -145,7 +145,7 @@ def crop_preserve_key_objects(input_folder, output_folder, model_path='yolov8n.p
         new_height = int(original_width / target_aspect_ratio)
 
         # YOLO inference
-        results = model.predict(frame, conf=0.3, verbose=False)
+        results = model.predict(frame, conf=0.1, verbose=False)
 
         # Initialize variables
         ball_detected = False