add person gathering

wangrong

2025-01-18 8b0e31121a9a29b92a41fc575138a2e9b4c12d8e

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#pragma once 
 
#include <opencv2/opencv.hpp> 
#include "NvInfer.h" 
#include "types.h" 
#include <map> 
 
 
void cuda_preprocess_init(int max_image_size); 
 
void cuda_preprocess_destroy(); 
 
void cuda_preprocess(uint8_t *src, int src_width, int src_height, float *dst, int dst_width, int dst_height, cudaStream_t stream); 
 
void cuda_batch_preprocess(std::vector<cv::Mat> &img_batch, float *dst, int dst_width, int dst_height, cudaStream_t stream);