41 arm_compute::DetectionPostProcessLayerInfo
info =
MakeInfo(desc);
43 const arm_compute::TensorInfo aclBoxEncodings =
44 armcomputetensorutils::BuildArmComputeTensorInfo(boxEncodings);
46 const arm_compute::TensorInfo aclScores =
47 armcomputetensorutils::BuildArmComputeTensorInfo(scores);
49 const arm_compute::TensorInfo aclAnchors =
50 armcomputetensorutils::BuildArmComputeTensorInfo(anchors);
52 arm_compute::TensorInfo aclDetectionBoxes =
53 armcomputetensorutils::BuildArmComputeTensorInfo(detectionBoxes);
55 arm_compute::TensorInfo aclDetectionClasses =
56 armcomputetensorutils::BuildArmComputeTensorInfo(detectionClasses);
58 arm_compute::TensorInfo aclDetectionScores =
59 armcomputetensorutils::BuildArmComputeTensorInfo(detectionScores);
61 arm_compute::TensorInfo aclNumDetections =
62 armcomputetensorutils::BuildArmComputeTensorInfo(numDetections);
64 return arm_compute::NEDetectionPostProcessLayer::validate(
80 m_Anchors = std::make_unique<arm_compute::Tensor>();
85 auto AclTensorRef = [](
ITensorHandle* tensor) -> arm_compute::ITensor&
87 return PolymorphicDowncast<IAclTensorHandle*>(tensor)->GetTensor();
93 arm_compute::ITensor& detectionBoxes = AclTensorRef(
m_Data.
m_Outputs[0]);
94 arm_compute::ITensor& detectionClasses = AclTensorRef(
m_Data.
m_Outputs[1]);
95 arm_compute::ITensor& detectionScores = AclTensorRef(
m_Data.
m_Outputs[2]);
96 arm_compute::ITensor& numDetections = AclTensorRef(
m_Data.
m_Outputs[3]);
98 m_Func.configure(&boxEncodings, &scores, m_Anchors.get(),
99 &detectionBoxes, &detectionClasses, &detectionScores, &numDetections,
const DetectionPostProcessQueueDescriptor m_Data
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
uint32_t m_DetectionsPerClass
Detections per classes, used in Regular NMS.
std::vector< float > boxEncodings({ 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f, 0.0f, -1.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f })
Copyright (c) 2020 ARM Limited.
LayerDescriptor m_Parameters
uint32_t m_MaxClassesPerDetection
Maximum numbers of classes per detection, used in Fast NMS.
uint32_t m_MaxDetections
Maximum numbers of detections.
arm_compute::Status NeonDetectionPostProcessValidate(const TensorInfo &boxEncodings, const TensorInfo &scores, const TensorInfo &anchors, const TensorInfo &detectionBoxes, const TensorInfo &detectionClasses, const TensorInfo &detectionScores, const TensorInfo &numDetections, const DetectionPostProcessDescriptor &desc)
float m_NmsIouThreshold
Intersection over union threshold.
NeonDetectionPostProcessWorkload(const DetectionPostProcessQueueDescriptor &descriptor, const WorkloadInfo &info)
uint32_t m_NumClasses
Number of classes.
const ConstCpuTensorHandle * m_Anchors
bool m_UseRegularNms
Use Regular NMS.
void InitializeArmComputeTensorData(arm_compute::Tensor &tensor, const ConstCpuTensorHandle *handle)
std::vector< float > scores({ 0.0f, 0.9f, 0.8f, 0.0f, 0.75f, 0.72f, 0.0f, 0.6f, 0.5f, 0.0f, 0.93f, 0.95f, 0.0f, 0.5f, 0.4f, 0.0f, 0.3f, 0.2f })
std::vector< ITensorHandle * > m_Outputs
Contains information about inputs and outputs to a layer.
float m_NmsScoreThreshold
NMS score threshold.
std::vector< ITensorHandle * > m_Inputs
arm_compute::DetectionPostProcessLayerInfo MakeInfo(const DetectionPostProcessDescriptor &desc)
virtual void Execute() const override
const TensorInfo & GetTensorInfo() const
std::vector< float > anchors({ 0.5f, 0.5f, 1.0f, 1.0f, 0.5f, 0.5f, 1.0f, 1.0f, 0.5f, 0.5f, 1.0f, 1.0f, 0.5f, 10.5f, 1.0f, 1.0f, 0.5f, 10.5f, 1.0f, 1.0f, 0.5f, 100.5f, 1.0f, 1.0f })