#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_