process_kernel_warp.hpp File Reference

#include <ReUseX/vision/tensor_rt/common/norm.hpp>
#include <ReUseX/vision/tensor_rt/kernels/postprocess.cuh>
#include <ReUseX/vision/tensor_rt/kernels/preprocess.cuh>

Include dependency graph for process_kernel_warp.hpp:

Go to the source code of this file.

Namespaces
namespace	ReUseX
namespace	ReUseX::vision
namespace	ReUseX::vision::tensor_rt

Functions
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)
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)
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)
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)