From 900980258b9c0bfbe0a5b452bdbf9124b359ff57 Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Mon, 16 Mar 2020 23:40:56 -0400
Subject: [PATCH] [common] (tf model) Add a minimum threshold for all objects
---
.../pipeline/objects_detectors/tf_model_objects_detector.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/common/polystar/common/pipeline/objects_detectors/tf_model_objects_detector.py b/common/polystar/common/pipeline/objects_detectors/tf_model_objects_detector.py
index 85817eb..9b6cda5 100644
--- a/common/polystar/common/pipeline/objects_detectors/tf_model_objects_detector.py
+++ b/common/polystar/common/pipeline/objects_detectors/tf_model_objects_detector.py
@@ -46,6 +46,7 @@ class TFModelObjectsDetector(ObjectsDetectorABC):
for box, class_id, score in zip(
output_dict["detection_boxes"], output_dict["detection_classes"], output_dict["detection_scores"]
)
+ if score >= 0.1
]
return objects
--
GitLab