onert::backend::cpu::ops::DetectionPostProcessLayer Class Reference

#include <DetectionPostProcessLayer.h>

Collaboration diagram for onert::backend::cpu::ops::DetectionPostProcessLayer:

Data Structures
struct	Allocations
struct	CenterSizeBox
struct	CornerBox
struct	DetectionPostProcessParameters

Public Types
enum	SelectionFormat { BOX_INDEX = 1 , CLASS_INDEX = 0 }

Public Member Functions
	DetectionPostProcessLayer ()
virtual	~DetectionPostProcessLayer ()
void	configure (DetectionPostProcessParameters parameters)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 33 of file DetectionPostProcessLayer.h.

Member Enumeration Documentation

SelectionFormat

enum onert::backend::cpu::ops::DetectionPostProcessLayer::SelectionFormat

Enumerator
BOX_INDEX
CLASS_INDEX

Definition at line 70 of file DetectionPostProcessLayer.h.

  {
    BOX_INDEX = 1,
    CLASS_INDEX = 0
  };

Constructor & Destructor Documentation

DetectionPostProcessLayer()

onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessLayer ( )

inline

Definition at line 82 of file DetectionPostProcessLayer.h.

                              : _parameters{}
  {
    // DO NOTHING
  }

~DetectionPostProcessLayer()

onert::backend::cpu::ops::DetectionPostProcessLayer::~DetectionPostProcessLayer ( )

virtual

Definition at line 303 of file DetectionPostProcessLayer.cc.

{ delete[] _allocations.selections_buffer; }

References onert::backend::cpu::ops::DetectionPostProcessLayer::Allocations::selections_buffer.

Member Function Documentation

configure()

void onert::backend::cpu::ops::DetectionPostProcessLayer::configure

(

DetectionPostProcessParameters

parameters

)

Definition at line 275 of file DetectionPostProcessLayer.cc.

{
  _parameters = std::move(parameters);
  _allocations.selections_buffer = new int[_parameters.max_detections * 2];
}

References onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::max_detections, and onert::backend::cpu::ops::DetectionPostProcessLayer::Allocations::selections_buffer.

run()

void onert::backend::cpu::ops::DetectionPostProcessLayer::run ( )

overridevirtual

Implements onert::exec::IFunction.

Definition at line 281 of file DetectionPostProcessLayer.cc.

{
  auto nbatches = (unsigned int)_parameters.boxes_descr[0];
  // no suport for batch other than 1( it's fine since tflite does not support
  // batch for postprocess either )
  assert(nbatches == 1);
 
  auto boxes_a = toArray<float>(_parameters.boxes_input->buffer(), _parameters.boxes_descr);
  auto scores_a = toArray<float>(_parameters.scores_input->buffer(), _parameters.scrores_descr);
 
  auto num_selected_a = ndarray::Array<float>(
    reinterpret_cast<float *>(_parameters.num_selections_output->buffer()), {nbatches});
 
  OutputArrays outputArrays(reinterpret_cast<CornerBox *>(_parameters.box_coords_output->buffer()),
                            reinterpret_cast<float *>(_parameters.box_scores_output->buffer()),
                            reinterpret_cast<float *>(_parameters.box_classes_output->buffer()),
                            reinterpret_cast<int *>(_parameters.num_selections_output->buffer()),
                            _parameters.max_detections);
 
  DetectionPostProcess(boxes_a, scores_a, num_selected_a, _parameters, _allocations, outputArrays);
}

References onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::box_classes_output, onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::box_coords_output, onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::box_scores_output, onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::boxes_descr, onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::boxes_input, onert::backend::ITensor::buffer(), onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::max_detections, onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::num_selections_output, onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::scores_input, and onert::backend::cpu::ops::DetectionPostProcessLayer::DetectionPostProcessParameters::scrores_descr.

Referenced by package.infer.session::inference().

---

The documentation for this class was generated from the following files:

• runtime/onert/backend/cpu/ops/DetectionPostProcessLayer.h
• runtime/onert/backend/cpu/ops/DetectionPostProcessLayer.cc