opennn::DetectionV8 Class Reference

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

Inheritance diagram for opennn::DetectionV8:

Public Member Functions
	DetectionV8 (const Shape &={}, const string &="detection_v8")
	DetectionV8 (const Shape &, Index reg_max, const string &="detection_v8")
Shape	get_output_shape () const override
Index	get_classes_number () const
DetectionHeadMetadata	get_detection_head_metadata () const noexcept override
void	set (const Shape &, const string &)
void	set (const Shape &, Index reg_max, const string &)
bool	accepts_input_rank (Index rank) const override
void	on_compute_dtype_changed () override
void	apply_input_shape (const Shape &new_input_shape) override
void	read_JSON_body (const Json *) override
void	write_JSON_body (JsonWriter &) const override
Public Member Functions inherited from opennn::Layer
virtual	~Layer ()=default
const string &	get_label () const noexcept
const string &	get_name () const
LayerType	get_type () const noexcept
void	set_input_shape (const Shape &new_input_shape)
virtual void	set_output_shape (const Shape &)
void	set_label (string new_label)
Index	get_parameters_number () const
const vector< Operator * > &	get_operators () const noexcept
virtual vector< TensorSpec >	get_parameter_specs () const
virtual vector< TensorSpec >	get_state_specs () const
virtual vector< TensorSpec >	get_forward_specs (Index batch_size) const
virtual vector< TensorSpec >	get_backward_specs (Index batch_size) const
virtual ForwardSlotKind	get_forward_slot_kind (size_t) const
virtual bool	is_forward_slot_inference_elidable (size_t, Device) const noexcept
virtual size_t	get_recomputable_forward_slot () const noexcept
virtual void	recompute_forward_slot (ForwardPropagation &, size_t)
virtual bool	backward_uses_forward_output () const noexcept
virtual bool	backward_uses_input (size_t) const noexcept
virtual bool	preserves_output_delta_during_backward () const noexcept
virtual bool	allows_input_delta_alias () const noexcept
virtual bool	folds_input_delta_addend (size_t) const noexcept
virtual Shape	get_input_shape () const noexcept
virtual ActivationFunction	get_output_activation () const
virtual Index	get_sources_number () const noexcept
virtual bool	allows_successors () const noexcept
virtual bool	is_recurrent () const noexcept
virtual bool	skip_for_pre_scaled_input () const noexcept
virtual bool	uses_sequence_position () const noexcept
virtual bool	allows_bf16_input_cast (size_t) const noexcept
Index	get_inputs_number () const noexcept
Index	get_outputs_number () const
virtual void	forward_propagate (ForwardPropagation &forward_propagation, size_t layer, ForwardPropagationMode pass)
virtual void	back_propagate (ForwardPropagation &forward_propagation, BackPropagation &back_propagation, size_t i) const
virtual void	from_JSON (const JsonDocument &)
virtual void	on_loaded ()
virtual void	load_state_from_JSON (const JsonDocument &)
virtual void	to_JSON (JsonWriter &) const
virtual string	write_expression (const vector< string > &, const vector< string > &) const
bool	get_is_trainable () const noexcept
void	set_is_trainable (bool trainable)
Type	get_compute_dtype () const noexcept
Device	get_compute_device () const noexcept
void	set_compute_dtype (Type new_compute_dtype)
void	set_compute_device (Device new_compute_device)
virtual float *	link_states (float *, Device)
float *	link_gradients (float *, Device)
vector< TensorView > &	get_parameter_views ()
const vector< TensorView > &	get_parameter_views () const noexcept
vector< TensorView > &	get_parameter_scales ()
const vector< TensorView > &	get_parameter_scales () const noexcept
vector< Operator::SlotQuantization >	get_parameter_quantization () const
virtual TiedWeight	get_tied_weight () const
virtual void	set_tied_weight (const TiedWeight &tied_weight)
void	redistribute_parameters_to_operators ()
Public Member Functions inherited from opennn::DetectionHeadEndpoint
virtual	~DetectionHeadEndpoint ()=default

Additional Inherited Members
Protected Types inherited from opennn::Layer
enum	Forward { Input , Output }
Protected Member Functions inherited from opennn::Layer
	Layer (LayerType type, Trainability trainability=Trainability::Trainable)
float *	link_views_to_operators (vector< TensorView > &, float *, vector< TensorSpec >(Operator::*specs_fn)() const, void(Operator::*link_fn)(span< const TensorView >), Device)
Static Protected Member Functions inherited from opennn::Layer
static bool	refresh_feature_storage (Buffer &storage, bool &dirty, Device device, Index features, Index columns, const function< void(float *)> &fill)
Protected Attributes inherited from opennn::Layer
string	label = "my_layer"
bool	is_trainable = true
Shape	input_shape
Type	compute_dtype = Type::FP32
Type	weights_dtype = Type::FP32
Device	compute_device = Device::CPU
vector< TensorView >	parameters
vector< TensorView >	parameter_scales
vector< TensorView >	states
vector< Operator * >	operators

Constructor & Destructor Documentation

DetectionV8() [1/2]

opennn::DetectionV8::DetectionV8	(	const Shape &	= {},
		const string &	= "detection_v8" )

DetectionV8() [2/2]

opennn::DetectionV8::DetectionV8	(	const Shape &	,
		Index	reg_max,
		const string &	= "detection_v8" )

Member Function Documentation

accepts_input_rank()

bool opennn::DetectionV8::accepts_input_rank ( Index rank ) const

inlineoverridevirtual

Reimplemented from opennn::Layer.

apply_input_shape()

void opennn::DetectionV8::apply_input_shape ( const Shape & new_input_shape )

inlineoverridevirtual

Reimplemented from opennn::Layer.

get_classes_number()

Index opennn::DetectionV8::get_classes_number ( ) const

inline

get_detection_head_metadata()

DetectionHeadMetadata opennn::DetectionV8::get_detection_head_metadata ( ) const

inlineoverridevirtualnoexcept

Implements opennn::DetectionHeadEndpoint.

get_output_shape()

Shape opennn::DetectionV8::get_output_shape ( ) const

inlineoverridevirtual

Implements opennn::Layer.

on_compute_dtype_changed()

void opennn::DetectionV8::on_compute_dtype_changed ( )

inlineoverridevirtual

Reimplemented from opennn::Layer.

read_JSON_body()

void opennn::DetectionV8::read_JSON_body ( const Json * )

overridevirtual

Reimplemented from opennn::Layer.

set() [1/2]

void opennn::DetectionV8::set	(	const Shape &	,
		const string &	)

set() [2/2]

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

write_JSON_body()

void opennn::DetectionV8::write_JSON_body ( JsonWriter & ) const

overridevirtual

Reimplemented from opennn::Layer.