Update IoU threshold to match Ultralytics default (#193)

* Update IoU threshold to match Ultralytics default

Signed-off-by: Mohammed Yasin <[email protected]>

* Update BasePredictor.swift

* Update IOU threshold in BasePredictorTests

Signed-off-by: Mohammed Yasin <[email protected]>

---------

Signed-off-by: Mohammed Yasin <[email protected]>
Co-authored-by: Glenn Jocher <[email protected]>

Author: Y-T-G

Sources/YOLO/BasePredictor.swift

@@ -253,10 +253,10 @@ public class BasePredictor: Predictor, @unchecked Sendable {
     confidenceThreshold = confidence
   }
 
-  /// The IoU (Intersection over Union) threshold for non-maximum suppression (default: 0.4).
+  /// The IoU (Intersection over Union) threshold for non-maximum suppression (default: 0.7).
   ///
   /// Used to filter overlapping detections during non-maximum suppression.
-  var iouThreshold = 0.4
+  var iouThreshold = 0.7
 
   /// Sets the IoU threshold for non-maximum suppression.
   ///

Tests/YOLOTests/BasePredictorTests.swift

@@ -16,7 +16,7 @@ class BasePredictorTests: XCTestCase {
     XCTAssertFalse(predictor.isModelLoaded)
     XCTAssertEqual(predictor.labels.count, 0)
     XCTAssertEqual(predictor.confidenceThreshold, 0.25, accuracy: 0.001)
-    XCTAssertEqual(predictor.iouThreshold, 0.4, accuracy: 0.001)
+    XCTAssertEqual(predictor.iouThreshold, 0.7, accuracy: 0.001)
     XCTAssertEqual(predictor.numItemsThreshold, 30)
     XCTAssertFalse(predictor.isUpdating)
     XCTAssertNil(predictor.currentBuffer)