fee7cbd0fe
* yolov8 * Update README.md * Update README.md * Update main.cpp * Delete yolov8/output directory * Delete yolov8/output directory
15 lines
434 B
C++
15 lines
434 B
C++
#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); |