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

Inheritance diagram for opennn::DetectionOperator:

Public Types
using	ClassActivation = DetectionClassActivation

Public Member Functions
void	set (const Shape &, const vector< array< float, 2 > > &)

void	forward_propagate (ForwardPropagation &, size_t, ForwardPropagationMode) override

void	back_propagate (ForwardPropagation &, BackPropagation &, size_t) const 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	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

ClassActivation	class_activation = ClassActivation::Softmax

vector< array< float, 2 > >	anchors

Buffer	device_anchors {Device::CUDA}

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 Typedef Documentation

◆ ClassActivation

using opennn::DetectionOperator::ClassActivation = DetectionClassActivation

Member Function Documentation

◆ back_propagate()

void opennn::DetectionOperator::back_propagate	(	ForwardPropagation &	,
		BackPropagation &	,
		size_t	) const

overridevirtual

Reimplemented from opennn::Operator.

◆ forward_propagate()

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

overridevirtual

Reimplemented from opennn::Operator.

◆ set()

void opennn::DetectionOperator::set	(	const Shape &	,
		const vector< array< float, 2 > > &	)

Member Data Documentation

◆ anchors

vector<array<float, 2> > opennn::DetectionOperator::anchors

◆ boxes_per_cell

Index opennn::DetectionOperator::boxes_per_cell = 0

◆ class_activation

ClassActivation opennn::DetectionOperator::class_activation = ClassActivation::Softmax

◆ classes_number

Index opennn::DetectionOperator::classes_number = 0

◆ device_anchors

Buffer opennn::DetectionOperator::device_anchors {Device::CUDA}

◆ grid_size

Index opennn::DetectionOperator::grid_size = 0

◆ grid_width

Index opennn::DetectionOperator::grid_width = 0

Public Types

Public Member Functions

Public Attributes