From 2fb1e5dd8c31ddfd6e25672adf61f4527c23213f Mon Sep 17 00:00:00 2001
From: Ninenox <nisit15@hotmail.com>
Date: Tue, 22 Aug 2023 12:52:22 +0700
Subject: [PATCH] check conf score negative

---
 models/torch_utils.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/models/torch_utils.py b/models/torch_utils.py
index f4e80ab..0946d62 100644
--- a/models/torch_utils.py
+++ b/models/torch_utils.py
@@ -64,6 +64,8 @@ def det_postprocess(data: Tuple[Tensor, Tensor, Tensor, Tensor]):
     iou_thres: float = 0.65
     num_dets, bboxes, scores, labels = data[0][0], data[1][0], data[2][
         0], data[3][0]
+    # check score negative
+    scores[scores < 0] = 1+scores[scores < 0]
     nums = num_dets.item()
     if nums == 0:
         return bboxes.new_zeros((0, 4)), scores.new_zeros(
@@ -71,6 +73,7 @@ def det_postprocess(data: Tuple[Tensor, Tensor, Tensor, Tensor]):
     bboxes = bboxes[:nums]
     scores = scores[:nums]
     labels = labels[:nums]
+    # add nms
     idx = nms(bboxes, scores, iou_thres)
     bboxes, scores, labels = bboxes[idx], scores[idx], labels[idx]
     return bboxes, scores, labels