Spaces:
Runtime error
Runtime error
oskarastrom
commited on
Commit
β’
a2da5d3
1
Parent(s):
41c67d5
Update inference.py
Browse files- inference.py +2 -3
inference.py
CHANGED
@@ -289,8 +289,7 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, b
|
|
289 |
|
290 |
|
291 |
if temp_frame is not None:
|
292 |
-
|
293 |
-
boost_frame(safe_frame, temp_frame, dt, power=boost_scale, decay=boost_decay)
|
294 |
|
295 |
pbar.update(1*batch_size)
|
296 |
|
@@ -298,7 +297,7 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, b
|
|
298 |
def boost_frame(safe_frame, base_frame, dt, power=1, decay=1):
|
299 |
safe_boxes = safe_frame[:, :4]
|
300 |
boxes = xywh2xyxy(base_frame[:, :4]) # center_x, center_y, width, height) to (x1, y1, x2, y2)β
|
301 |
-
ious = box_iou(boxes, safe_boxes)
|
302 |
score = torch.matmul(ious, safe_frame[:, 4])
|
303 |
# score = iou(safe_box, base_box) * confidence(safe_box)
|
304 |
base_frame[:, 4] *= 1 + power*(score)*math.exp(-decay*(dt*dt-1))
|
|
|
289 |
|
290 |
|
291 |
if temp_frame is not None:
|
292 |
+
boost_frame(safe_frame, temp_frame, dt, power=boost_scale, decay=boost_decay)
|
|
|
293 |
|
294 |
pbar.update(1*batch_size)
|
295 |
|
|
|
297 |
def boost_frame(safe_frame, base_frame, dt, power=1, decay=1):
|
298 |
safe_boxes = safe_frame[:, :4]
|
299 |
boxes = xywh2xyxy(base_frame[:, :4]) # center_x, center_y, width, height) to (x1, y1, x2, y2)β
|
300 |
+
ious = box_iou(boxes.double(), safe_boxes)
|
301 |
score = torch.matmul(ious, safe_frame[:, 4])
|
302 |
# score = iou(safe_box, base_box) * confidence(safe_box)
|
303 |
base_frame[:, 4] *= 1 + power*(score)*math.exp(-decay*(dt*dt-1))
|