Filter invalid iseg boxes in cpp to match python

openvinotoolkit · Feb 22, 2024 · cd91bdb · cd91bdb
1 parent c3998dc
commit cd91bdb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/model_api/cpp/models/src/instance_segmentation.cpp b/model_api/cpp/models/src/instance_segmentation.cpp
@@ -356,6 +356,11 @@ std::unique_ptr<ResultBase> MaskRCNNModel::postprocess(InferenceResult& infResul
         obj.height = clamp(
             round((boxes[i * objectSize + 3] - padTop) * invertedScaleY - obj.y),
             0.f, floatInputImgHeight);
+
+        if (obj.height * obj.width <= 1) {
+            continue;
+        }
+
         cv::Mat raw_cls_mask{masks_size, CV_32F, masks + masks_size.area() * i};
         cv::Mat resized_mask;
         if (postprocess_semantic_masks || has_feature_vector_name) {