From 0c24764ff3022d0013b5b7fba1a2b3e1e560659c Mon Sep 17 00:00:00 2001
From: tanmodong <93513016+tanmodong@users.noreply.github.com>
Date: Thu, 22 Dec 2022 13:40:12 +0900
Subject: [PATCH] score based instance filter added

originally there is no bounding box score filtration in video inferencemode.
Added score based instance filter in instance mode.
---
 diffusiondet/predictor.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/diffusiondet/predictor.py b/diffusiondet/predictor.py
index 94489bf..3fc8843 100644
--- a/diffusiondet/predictor.py
+++ b/diffusiondet/predictor.py
@@ -101,6 +101,8 @@ def process_predictions(frame, predictions):
                 )
             elif "instances" in predictions:
                 predictions = predictions["instances"].to(self.cpu_device)
+                new_instances = predictions[predictions.scores > self.threshold]
+                predictions = new_instances
                 vis_frame = video_visualizer.draw_instance_predictions(frame, predictions)
             elif "sem_seg" in predictions:
                 vis_frame = video_visualizer.draw_sem_seg(