Revisit detections filtering

Author: sovrasov

model_api/python/model_api/models/detection_model.py

@@ -105,24 +105,54 @@ def _resize_detections(self, detections, meta):
                 pad_top = (self.h - round(input_img_height / inverted_scale_y)) // 2
 
         for detection in detections:
-            detection.xmin = min(
-                max(round((detection.xmin * self.w - pad_left) * inverted_scale_x), 0),
-                input_img_widht,
+            detection.xmin = round(
+                min(
+                    max((detection.xmin * self.w - pad_left) * inverted_scale_x, 0),
+                    input_img_widht,
+                )
             )
-            detection.ymin = min(
-                max(round((detection.ymin * self.h - pad_top) * inverted_scale_y), 0),
-                input_img_height,
+            detection.ymin = round(
+                min(
+                    max((detection.ymin * self.h - pad_top) * inverted_scale_y, 0),
+                    input_img_height,
+                )
             )
-            detection.xmax = min(
-                max(round((detection.xmax * self.w - pad_left) * inverted_scale_x), 0),
-                input_img_widht,
+            detection.xmax = round(
+                min(
+                    max((detection.xmax * self.w - pad_left) * inverted_scale_x, 0),
+                    input_img_widht,
+                )
             )
-            detection.ymax = min(
-                max(round((detection.ymax * self.h - pad_top) * inverted_scale_y), 0),
-                input_img_height,
+            detection.ymax = round(
+                min(
+                    max((detection.ymax * self.h - pad_top) * inverted_scale_y, 0),
+                    input_img_height,
+                )
             )
         return detections
 
+    def _filter_detections(self, detections, box_area_threshold=0):
+        """Filters detections by confidence threshold and box size threshold
+
+        Args:
+            detections (List[Detection]): list of detections with coordinates in normalized form
+            box_area_threshold (float): minimal area of the bounding to be considered
+
+        Returns:
+            - list of detections with confidence above the threshold
+        """
+        filtered_detections = []
+        for detection in detections:
+            if (
+                detection.score < self.confidence_threshold
+                or (detection.xmax - detection.xmin) * (detection.ymax - detection.ymin)
+                < box_area_threshold
+            ):
+                continue
+            filtered_detections.append(detection)
+
+        return filtered_detections
+
     def _add_label_names(self, detections):
         """Adds labels names to detections if they are available
 

model_api/python/model_api/models/ssd.py

@@ -41,6 +41,7 @@ def preprocess(self, inputs):
     def postprocess(self, outputs, meta):
         detections = self._parse_outputs(outputs)
         detections = self._resize_detections(detections, meta)
+        detections = self._filter_detections(detections, 1)
         detections = self._add_label_names(detections)
         return DetectionResult(
             detections,