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

Inheritance diagram for opennn::DetectionV8Operator:

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

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	classes_number = 0

Index	reg_max = 1

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

◆ back_propagate()

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

overridevirtual

Reimplemented from opennn::Operator.

◆ forward_propagate()

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

overridevirtual

Reimplemented from opennn::Operator.

◆ set()

void opennn::DetectionV8Operator::set	(	const Shape &	,
		Index	reg_max )

Member Data Documentation

◆ classes_number

Index opennn::DetectionV8Operator::classes_number = 0

◆ grid_size

Index opennn::DetectionV8Operator::grid_size = 0

◆ grid_width

Index opennn::DetectionV8Operator::grid_width = 0

◆ reg_max

Index opennn::DetectionV8Operator::reg_max = 1

Public Member Functions

Public Attributes