e73bffcd25
* tensorrt-yolov9 * format change * change format * Update block.cpp * format: add space * update block.cpp: add space * update config.h --------- Co-authored-by: Wang Xinyu <wangxinyu_es@163.com>
16 lines
522 B
C++
16 lines
522 B
C++
#pragma once
|
|
|
|
#include <cuda_runtime.h>
|
|
#include <cstdint>
|
|
#include <opencv2/opencv.hpp>
|
|
|
|
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);
|
|
|