Update inference.py

inference.py

@@ -289,8 +289,7 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, b
                     
 
                     if temp_frame is not None:
-                        with torch.no_grad():
-                            boost_frame(safe_frame, temp_frame, dt, power=boost_scale, decay=boost_decay)
+                        boost_frame(safe_frame, temp_frame, dt, power=boost_scale, decay=boost_decay)
 
             pbar.update(1*batch_size)
           
@@ -298,7 +297,7 @@ 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, safe_boxes)
+    ious = box_iou(boxes.double(), safe_boxes)
     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))