reusex::vision::tensor_rt::TensorRTSam3p1 Class Reference

#include <Sam3p1.hpp>

Inheritance diagram for reusex::vision::tensor_rt::TensorRTSam3p1:

Collaboration diagram for reusex::vision::tensor_rt::TensorRTSam3p1:

Public Member Functions
	TensorRTSam3p1 (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 &memory_encoder_path, const std::string &memory_attention_path, const std::string &tokenizer_path, const std::filesystem::path &meta_path, int gpu_id)
void	reset () override
IDataset::Pair	step (const IDataset::Pair &in) override
Public Member Functions inherited from reusex::vision::IVideoModel
virtual	~IVideoModel ()=default

Static Public Member Functions
static std::unique_ptr< TensorRTSam3p1 >	create (const std::filesystem::path &model_path)

Protected Member Functions
bool	load_engines ()

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

Detailed Description

Definition at line 46 of file Sam3p1.hpp.

Constructor & Destructor Documentation

TensorRTSam3p1()

reusex::vision::tensor_rt::TensorRTSam3p1::TensorRTSam3p1	(	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 &	memory_encoder_path,
		const std::string &	memory_attention_path,
		const std::string &	tokenizer_path,
		const std::filesystem::path &	meta_path,
		int	gpu_id )

Member Function Documentation

create()

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

static

load_bytes_from_file()

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

staticprotected

load_engines()

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

protected

reset()

void reusex::vision::tensor_rt::TensorRTSam3p1::reset ( )

overridevirtual

Implements reusex::vision::IVideoModel.

step()

IDataset::Pair reusex::vision::tensor_rt::TensorRTSam3p1::step ( const IDataset::Pair & in )

overridevirtual

Implements reusex::vision::IVideoModel.

---

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

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