41 lines
1.0 KiB
C++
41 lines
1.0 KiB
C++
#ifndef _PRE_PROCESS_H_
|
|
#define _PRE_PROCESS_H_
|
|
|
|
#include <cuda_runtime.h>
|
|
#include <cstdint>
|
|
|
|
#include <opencv2/opencv.hpp>
|
|
#include <opencv2/core/core.hpp>
|
|
#include <opencv2/highgui/highgui.hpp>
|
|
#include <opencv2/imgproc/imgproc.hpp>
|
|
|
|
struct AffineMatrix{
|
|
float value[6];
|
|
};
|
|
|
|
void yolov5_detect_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_detect_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);
|
|
|
|
cv::Mat preprocess_img(cv::Mat& img, int input_w, int input_h);
|
|
|
|
#endif //END OF _PRE_PROCESS_H_
|
|
|
|
|