FIX: add clamp for iou_score

ShiqiYu · Dec 23, 2020 · 2f4001b · 2f4001b
1 parent 67489d6
commit 2f4001b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/facedetectcnn.cpp b/src/facedetectcnn.cpp
@@ -871,7 +871,15 @@ bool detection_output(const CDataBlob<float> * priorbox,
         float cls_score = pConf[i + 1];
         int face_idx = i / 2;
         float iou_score = pIoU[face_idx];
+        // clamp
+        if (iou_score < 0.f) {
+            iou_score = 0.f;
+        }
+        else if (iou_score > 1.f) {
+            iou_score = 1.f;
+        }
         float conf = sqrtf(cls_score * iou_score);
+
         if(conf > confidence_threshold)
         {
             float fBox_x1 = pPriorBox[face_idx * 4];