64da26ef5d
* yolov7 v0.1 * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * Update yololayer.h * Update yololayer.h * Delete yolov7.engine * Delete yolov7-tiny.engine * Delete yolov7/output directory * Delete dirent.h * Update README.md * Update README.md * Update README.md * Update README.md
17 lines
357 B
C++
17 lines
357 B
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);
|
|
#endif // __PREPROCESS_H
|