ReUseX::vision::tensor_rt Namespace Reference

Namespaces
namespace	affine
namespace	nv
namespace	cpm
namespace	tensor
namespace	norm_image
namespace	TensorRT

Classes
class	TensorRTBackend
class	AutoDevice
struct	TensorRTData
class	TensorRTDataset
class	TensorRTSam3
struct	Sam3PromptUnit
struct	Sam3Input

Typedefs
using	BoxPrompt = std::pair<std::string, std::array<float, 4>>

Functions
template<typename... Args>
constexpr void	Assertf (bool cond, const char *fmt, Args... args)
constexpr void	Asserts (bool cond, const char *s)
constexpr void	Assert (bool cond)
void	warp_affine_bilinear_and_normalize_plane (uint8_t *src, int src_line_size, int src_width, int src_height, float *dst, int dst_width, int dst_height, float *matrix_2_3, uint8_t const_value, const norm_image::Norm &norm, cudaStream_t stream)
void	warp_affine_bilinear_single_channel_plane (float *src, int src_line_size, int src_width, int src_height, float *dst, int dst_width, int dst_height, float *matrix_2_3, float const_value, cudaStream_t stream)
void	warp_affine_bilinear_single_channel_mask_plane (float *src, int src_line_size, int src_width, int src_height, uint8_t *dst, int dst_width, int dst_height, float *matrix_2_3, uint8_t const_value, cudaStream_t stream)
void	sam3_postprocess_plane (const float *pred_masks_gpu, const float *pred_boxes_gpu, const float *pred_logits_gpu, const float *presence_logits_gpu, float *filtered_boxes_gpu, int *filtered_indices_gpu, float *filtered_scores_gpu, int *filtered_count_gpu, int num_queries, int mask_height, int mask_width, int original_image_width, int original_image_height, float confidence_threshold, cudaStream_t stream)

Typedef Documentation

BoxPrompt

using ReUseX::vision::tensor_rt::BoxPrompt = std::pair<std::string, std::array<float, 4>>

Definition at line 11 of file Sam3Type.hpp.

Function Documentation

Assert()

void ReUseX::vision::tensor_rt::Assert ( bool cond )

constexpr

Definition at line 112 of file check.hpp.

Assertf()

template<typename... Args>

void ReUseX::vision::tensor_rt::Assertf ( bool cond, const char * fmt, Args... args )

constexpr

Definition at line 94 of file check.hpp.

Asserts()

void ReUseX::vision::tensor_rt::Asserts ( bool cond, const char * s )

constexpr

Definition at line 104 of file check.hpp.

sam3_postprocess_plane()

void ReUseX::vision::tensor_rt::sam3_postprocess_plane	(	const float *	pred_masks_gpu,
		const float *	pred_boxes_gpu,
		const float *	pred_logits_gpu,
		const float *	presence_logits_gpu,
		float *	filtered_boxes_gpu,
		int *	filtered_indices_gpu,
		float *	filtered_scores_gpu,
		int *	filtered_count_gpu,
		int	num_queries,
		int	mask_height,
		int	mask_width,
		int	original_image_width,
		int	original_image_height,
		float	confidence_threshold,
		cudaStream_t	stream )

warp_affine_bilinear_and_normalize_plane()

void ReUseX::vision::tensor_rt::warp_affine_bilinear_and_normalize_plane	(	uint8_t *	src,
		int	src_line_size,
		int	src_width,
		int	src_height,
		float *	dst,
		int	dst_width,
		int	dst_height,
		float *	matrix_2_3,
		uint8_t	const_value,
		const norm_image::Norm &	norm,
		cudaStream_t	stream )

warp_affine_bilinear_single_channel_mask_plane()

void ReUseX::vision::tensor_rt::warp_affine_bilinear_single_channel_mask_plane	(	float *	src,
		int	src_line_size,
		int	src_width,
		int	src_height,
		uint8_t *	dst,
		int	dst_width,
		int	dst_height,
		float *	matrix_2_3,
		uint8_t	const_value,
		cudaStream_t	stream )

warp_affine_bilinear_single_channel_plane()

void ReUseX::vision::tensor_rt::warp_affine_bilinear_single_channel_plane	(	float *	src,
		int	src_line_size,
		int	src_width,
		int	src_height,
		float *	dst,
		int	dst_width,
		int	dst_height,
		float *	matrix_2_3,
		float	const_value,
		cudaStream_t	stream )