fix: verify confidence tc value

PARC · Jul 22, 2022 · da7487a · da7487a · fritz-ebner · Jul 22, 2022
1 parent 8dd212b
commit da7487a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ALTOAI-Detect/MLModels/YOLO.swift b/ALTOAI-Detect/MLModels/YOLO.swift
@@ -55,7 +55,7 @@ class YOLO {
 
             let (detectedClass, bestClassScore) = class_probs.argmax()
 
-            let confidenceInClass = bestClassScore * confidence
+            let confidenceInClass = tc
 
             if confidenceInClass >= confidenceThreshold {
                 let rect = CGRect(x: CGFloat(tx)  - (CGFloat(tw)/2), y: CGFloat(ty) - (CGFloat(th)/2),