#include <opennn/network/layers/non_max_suppression_layer.h>

Inheritance diagram for opennn::NonMaxSuppressionOperator:

Public Member Functions
void	set (const Shape &, Index, float, float)

void	forward_propagate (ForwardPropagation &, size_t, ForwardPropagationMode) override

Public Member Functions inherited from opennn::Operator
virtual	~Operator ()=default

virtual vector< TensorSpec >	parameter_specs () const

virtual vector< TensorSpec >	state_specs () const

virtual vector< SlotQuantization >	parameter_quantization () const

virtual vector< ParameterSlot >	parameter_slots ()

virtual void	link_parameters (span< const TensorView > views)

virtual void	link_gradients (span< const TensorView > views)

virtual void	link_states (span< const TensorView >)

virtual void	link_parameter_scales (span< const TensorView >)

bool	link_slots (span< const TensorView > views, TensorView ParameterSlot::member)

virtual void	initialize_states ()

virtual void	set_weights_dtype (Type new_weights_dtype)

virtual void	set_parameters_random ()

virtual void	set_parameters_glorot ()

virtual void	set_parameters_pytorch ()

virtual void	back_propagate (ForwardPropagation &, BackPropagation &, size_t) const

virtual void	to_JSON (JsonWriter &) const

virtual void	from_JSON (const Json *)

virtual void	load_state_from_JSON (const Json *)

TensorView &	get_input (ForwardPropagation &forward_propagation, size_t layer, size_t slot_index=0) const noexcept

vector< TensorView > &	get_inputs (ForwardPropagation &forward_propagation, size_t layer) const noexcept

TensorView &	get_output (ForwardPropagation &forward_propagation, size_t layer, size_t slot_index=0) const noexcept

TensorView &	get_output_delta (BackPropagation &back_propagation, size_t layer, size_t slot_index=0) const noexcept

TensorView &	get_input_delta (BackPropagation &back_propagation, size_t layer, size_t slot_index=0) const noexcept

Public Attributes
Index	grid_size = 0

Index	grid_width = 0

Index	boxes_per_cell = 0

Index	classes_number = 0

float	confidence_threshold = 0.5f

float	iou_threshold = 0.4f

Public Attributes inherited from opennn::Operator
Type	compute_dtype = Type::FP32

Type	weights_dtype = Type::FP32

vector< size_t >	input_slots = {0}

vector< size_t >	output_slots = {1}

vector< size_t >	input_delta_slots = {1}

vector< size_t >	output_delta_slots = {0}

Member Function Documentation

◆ forward_propagate()

void opennn::NonMaxSuppressionOperator::forward_propagate	(	ForwardPropagation &	,
		size_t	,
		ForwardPropagationMode	)

overridevirtual

Reimplemented from opennn::Operator.

◆ set()

void opennn::NonMaxSuppressionOperator::set	(	const Shape &	,
		Index	,
		float	,
		float	)

Member Data Documentation

◆ boxes_per_cell

Index opennn::NonMaxSuppressionOperator::boxes_per_cell = 0

◆ classes_number

Index opennn::NonMaxSuppressionOperator::classes_number = 0

◆ confidence_threshold

float opennn::NonMaxSuppressionOperator::confidence_threshold = 0.5f

◆ grid_size

Index opennn::NonMaxSuppressionOperator::grid_size = 0

◆ grid_width

Index opennn::NonMaxSuppressionOperator::grid_width = 0

◆ iou_threshold

float opennn::NonMaxSuppressionOperator::iou_threshold = 0.4f

Public Member Functions

Public Attributes