Update inference.py

inference.py

@@ -297,8 +297,8 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, b
 def boost_frame(safe_frame, base_frame, dt, power=1, decay=1):
     safe_boxes = safe_frame[:, :4]
     boxes = xywh2xyxy(base_frame[:, :4])  # center_x, center_y, width, height) to (x1, y1, x2, y2)≈
-    ious = box_iou(boxes.double(), safe_boxes)
-    score = torch.matmul(ious.double(), safe_frame[:, 4])
+    ious = box_iou(boxes.double(), safe_boxes).float()
+    score = torch.matmul(ious, safe_frame[:, 4])
     # score = iou(safe_box, base_box) * confidence(safe_box)
     base_frame[:, 4] *= 1 + power*(score)*math.exp(-decay*(dt*dt-1))
     return base_frame