generated from zhangwei/Matrixai
34 lines
1.2 KiB
C++
34 lines
1.2 KiB
C++
#ifndef __PREPROCESS_H
|
|
#define __PREPROCESS_H
|
|
|
|
#include <cuda_runtime.h>
|
|
#include <cstdint>
|
|
|
|
|
|
struct AffineMatrix{
|
|
float value[6];
|
|
};
|
|
|
|
|
|
void preprocess_kernel_img(uint8_t* src, int src_width, int src_height,
|
|
float* dst, int dst_width, int dst_height,
|
|
cudaStream_t stream);
|
|
|
|
void yolov5_preprocess_kernel_img(uint8_t* src, int src_width, int src_height,
|
|
float* dst, int dst_width, int dst_height,
|
|
cudaStream_t stream);
|
|
|
|
void yolov5_clear_preprocess_kernel_img(uint8_t* src, int src_width, int src_height,
|
|
float* dst, int dst_width, int dst_height,
|
|
cudaStream_t stream);
|
|
|
|
void yolov5_classify_preprocess_kernel_img(uint8_t* src, int src_width, int src_height,
|
|
float* dst, int dst_width, int dst_height,
|
|
cudaStream_t stream);
|
|
|
|
void retinanet_classify_preprocess_kernel_img(uint8_t* src, int src_width, int src_height,
|
|
float* dst, int dst_width, int dst_height,
|
|
cudaStream_t stream);
|
|
|
|
#endif // __PREPROCESS_H
|