From 8bcd75eed8aec1ca562beed53cf43cf9a4ed982e Mon Sep 17 00:00:00 2001 From: Johan MEJIA Date: Mon, 22 Nov 2021 17:10:11 +0100 Subject: [PATCH] update threshold --- alonet/detr_panoptic/eval_on_coco.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/alonet/detr_panoptic/eval_on_coco.py b/alonet/detr_panoptic/eval_on_coco.py index e4d8d781..f2bf92ab 100644 --- a/alonet/detr_panoptic/eval_on_coco.py +++ b/alonet/detr_panoptic/eval_on_coco.py @@ -30,7 +30,7 @@ def main(args): for it, data in enumerate(coco_loader.val_dataloader()): frame = Frame.batch_list(data).to(device) - pred_boxes, pred_masks = lit_panoptic.inference(lit_panoptic(frame)) + pred_boxes, pred_masks = lit_panoptic.inference(lit_panoptic(frame, threshold=0.85)) pred_boxes, pred_masks = pred_boxes[0], pred_masks[0] gt_boxes = frame.boxes2d[0] # Get gt boxes as BoundingBoxes2D. gt_masks = frame.segmentation[0] # Get gt masks as Mask