From 22dce9a5e84a3f3e94d027e8b7ab2034effb1ddb Mon Sep 17 00:00:00 2001 From: Vladislav Sovrasov Date: Sat, 8 Jul 2023 03:47:38 +0900 Subject: [PATCH] Del extra masks check --- .../python/openvino/model_api/tilers/instance_segmentation.py | 4 +--- 1 file changed, 1 insertion(+), 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 index 49af48ac..bd6bd451 100644 --- 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)