Del extra masks check

openvinotoolkit · Jul 7, 2023 · 22dce9a · 22dce9a
1 parent 97ac9d4
commit 22dce9a
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/model_api/python/openvino/model_api/tilers/instance_segmentation.py b/model_api/python/openvino/model_api/tilers/instance_segmentation.py
@@ -123,9 +123,7 @@ def _merge_results(self, results, shape):
             detections_array, keep_idxs = _multiclass_nms(
                 detections_array, max_num=self.max_pred_number
             )
-
-        if masks:
-            masks = [masks[keep_idx] for keep_idx in keep_idxs]
+        masks = [masks[keep_idx] for keep_idx in keep_idxs]
 
         merged_vector = (
             np.mean(feature_vectors, axis=0) if feature_vectors else np.ndarray(0)