YoloModel Class Reference

#include <YoloModel.hpp>

Classes
struct	BoundingBox

Public Member Functions
	YoloModel ()=default
tl::expected< bool, std::string >	create (std::vector< int8_t > &&model_data, std::vector< std::string > labels, std::string_view gpu_delegate_serialization_dir, std::string_view model_token, TfLiteLogWarningCallback log_warning_callback, TfLiteLogErrorCallback log_error_callback, bool enable_npu, std::string skel_directory_dir)
tl::expected< std::vector< BoundingBox >, std::string >	run (FloatTensorBuffer< FloatTensorFormat::ImageRGB255 > &input)
std::vector< BoundingBox >	best_box (std::span< const float > array) const
std::span< const int >	get_input_shape ()
std::span< const int >	get_output_shape ()

Public Attributes
size_t	num_channel = 0
size_t	num_elements = 0

Constructor & Destructor Documentation

YoloModel()

YoloModel::YoloModel ( )

default

Member Function Documentation

best_box()

std::vector< YoloModel::BoundingBox > YoloModel::best_box ( std::span< const float > array ) const

nodiscard

create()

tl::expected< bool, std::string > YoloModel::create	(	std::vector< int8_t > &&	model_data,
		std::vector< std::string >	labels,
		std::string_view	gpu_delegate_serialization_dir,
		std::string_view	model_token,
		TfLiteLogWarningCallback	log_warning_callback,
		TfLiteLogErrorCallback	log_error_callback,
		bool	enable_npu,
		std::string	skel_directory_dir )

get_input_shape()

std::span< const int > YoloModel::get_input_shape

(

)

get_output_shape()

std::span< const int > YoloModel::get_output_shape

(

)

run()

tl::expected< std::vector< YoloModel::BoundingBox >, std::string > YoloModel::run

(

FloatTensorBuffer< FloatTensorFormat::ImageRGB255 > &

input

)

Member Data Documentation

num_channel

size_t YoloModel::num_channel = 0

num_elements

size_t YoloModel::num_elements = 0