CPP Function to generate the detection output based on center size encoded boxes, class prediction an...
NEDetectionPostProcessLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Constructor.
void configure(const ITensor *input_box_encoding, const ITensor *input_score, const ITensor *input_anchors, ITensor *output_boxes, ITensor *output_classes, ITensor *output_scores, ITensor *num_detection, DetectionPostProcessLayerInfo info=DetectionPostProcessLayerInfo())
Configure the detection output layer NE function.
NEDetectionPostProcessLayer & operator=(const NEDetectionPostProcessLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void run() override
Run the kernels contained in the function.
Detection Output layer info.
NE Function to generate the detection output based on center size encoded boxes, class prediction and...
static Status validate(const ITensorInfo *input_box_encoding, const ITensorInfo *input_class_score, const ITensorInfo *input_anchors, ITensorInfo *output_boxes, ITensorInfo *output_classes, ITensorInfo *output_scores, ITensorInfo *num_detection, DetectionPostProcessLayerInfo info=DetectionPostProcessLayerInfo())
Static function to check if given info will lead to a valid configuration of NEDetectionPostProcessLa...
~NEDetectionPostProcessLayer()=default
Default destructor.