diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-09-27 19:07:15 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-10-01 15:04:42 +0000 |
| commit | b94c368a39ba6518d843c66a156f825a2c66af60 (patch) | |
| tree | 384c4e4ed83389f300f491a63cd70a23e646babe /arm_compute | |
| parent | 3d8fe497fc76ec6ad265c03fe02e29ed2ddf2d93 (diff) | |
| download | ComputeLibrary-b94c368a39ba6518d843c66a156f825a2c66af60.tar.gz | |
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 <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2000
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Isabella Gottardi <isabella.gottardi@arm.com>
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/core/CPP/kernels/CPPNonMaximumSuppressionKernel.h | 5 | ||||
| -rw-r--r-- | arm_compute/runtime/CPP/functions/CPPDetectionPostProcessLayer.h | 9 |
2 files changed, 0 insertions, 14 deletions
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<float> _scores_above_thd_vector; - std::vector<int> _indices_above_thd_vector; - std::vector<bool> _visited; - std::vector<unsigned int> _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<int> _result_idx_boxes_after_nms; - std::vector<int> _result_classes_after_nms; - std::vector<float> _result_scores_after_nms; - std::vector<unsigned int> _sorted_indices; - - // Temporary values - std::vector<float> _box_scores; }; } // namespace arm_compute #endif /* __ARM_COMPUTE_CPP_DETECTION_POSTPROCESS_H__ */ |