reusex::vision::tensor_rt::TensorRTSam3 Class Reference

#include <Sam3.hpp>

Inheritance diagram for reusex::vision::tensor_rt::TensorRTSam3:

Collaboration diagram for reusex::vision::tensor_rt::TensorRTSam3:

Public Member Functions
	TensorRTSam3 (const std::string &vision_encoder_path, const std::string &text_encoder_path, const std::string &geometry_encoder_path, const std::string &decoder_path, const std::string &tokenizer_path, int gpu_id)
std::vector< IDataset::Pair >	forward (const std::span< IDataset::Pair > &input) override
Public Member Functions inherited from reusex::vision::IModel
virtual	~IModel ()=default

Static Public Member Functions
static std::unique_ptr< TensorRTSam3 >	create (const std::filesystem::path &model_path)
Static Public Member Functions inherited from reusex::vision::IModel
static std::unique_ptr< IModel >	create (const std::filesystem::path &model_path, bool use_gpu=false)

Protected Member Functions
bool	load_engines ()
bool	setup_geometry_input (const cv::Mat &image, const std::string &label, const std::vector< std::pair< std::string, std::array< float, 4 > > > &boxes)

Static Protected Member Functions
static std::string	load_bytes_from_file (const std::string &file_path)

Detailed Description

Definition at line 29 of file Sam3.hpp.

Constructor & Destructor Documentation

TensorRTSam3()

reusex::vision::tensor_rt::TensorRTSam3::TensorRTSam3	(	const std::string &	vision_encoder_path,
		const std::string &	text_encoder_path,
		const std::string &	geometry_encoder_path,
		const std::string &	decoder_path,
		const std::string &	tokenizer_path,
		int	gpu_id )

Member Function Documentation

create()

std::unique_ptr< TensorRTSam3 > reusex::vision::tensor_rt::TensorRTSam3::create ( const std::filesystem::path & model_path )

static

forward()

std::vector< IDataset::Pair > reusex::vision::tensor_rt::TensorRTSam3::forward ( const std::span< IDataset::Pair > & input )

overridevirtual

Implements reusex::vision::IModel.

load_bytes_from_file()

std::string reusex::vision::tensor_rt::TensorRTSam3::load_bytes_from_file ( const std::string & file_path )

staticprotected

load_engines()

bool reusex::vision::tensor_rt::TensorRTSam3::load_engines ( )

protected

setup_geometry_input()

bool reusex::vision::tensor_rt::TensorRTSam3::setup_geometry_input ( const cv::Mat & image, const std::string & label, const std::vector< std::pair< std::string, std::array< float, 4 > > > & boxes )

protected

---

The documentation for this class was generated from the following file:

• libs/reusex/include/vision/tensor_rt/Sam3.hpp