From b94c368a39ba6518d843c66a156f825a2c66af60 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 27 Sep 2019 19:07:15 +0100 Subject: COMPMID-2645: Detection Post-Process fixes -Uses temporary buffers withing run() to avoid keeping state -Reworks and cleans the implementation Change-Id: I34b5bad7f8a07e284214ec61de9cf3a2eb1f608c Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/2000 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Isabella Gottardi --- arm_compute/core/CPP/kernels/CPPNonMaximumSuppressionKernel.h | 5 ----- arm_compute/runtime/CPP/functions/CPPDetectionPostProcessLayer.h | 9 --------- 2 files changed, 14 deletions(-) (limited to 'arm_compute') diff --git a/arm_compute/core/CPP/kernels/CPPNonMaximumSuppressionKernel.h b/arm_compute/core/CPP/kernels/CPPNonMaximumSuppressionKernel.h index dd035d5cb1..351b9842c7 100644 --- a/arm_compute/core/CPP/kernels/CPPNonMaximumSuppressionKernel.h +++ b/arm_compute/core/CPP/kernels/CPPNonMaximumSuppressionKernel.h @@ -92,11 +92,6 @@ private: float _iou_threshold; unsigned int _num_boxes; - - std::vector _scores_above_thd_vector; - std::vector _indices_above_thd_vector; - std::vector _visited; - std::vector _sorted_indices; }; } // namespace arm_compute #endif /* __ARM_COMPUTE_CPP_NONMAXIMUMSUPPRESSIONKERNEL_LAYER_H__ */ diff --git a/arm_compute/runtime/CPP/functions/CPPDetectionPostProcessLayer.h b/arm_compute/runtime/CPP/functions/CPPDetectionPostProcessLayer.h index c13def67c7..1c918d220c 100644 --- a/arm_compute/runtime/CPP/functions/CPPDetectionPostProcessLayer.h +++ b/arm_compute/runtime/CPP/functions/CPPDetectionPostProcessLayer.h @@ -109,15 +109,6 @@ private: Tensor _selected_indices; Tensor _class_scores; const ITensor *_input_scores_to_use; - - // Intermediate results - std::vector _result_idx_boxes_after_nms; - std::vector _result_classes_after_nms; - std::vector _result_scores_after_nms; - std::vector _sorted_indices; - - // Temporary values - std::vector _box_scores; }; } // namespace arm_compute #endif /* __ARM_COMPUTE_CPP_DETECTION_POSTPROCESS_H__ */ -- cgit v1.2.1