You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.7 KiB
Plaintext
43 lines
1.7 KiB
Plaintext
#pragma once
|
|
#include "common_include.h"
|
|
#include "utils.h"
|
|
|
|
#define checkRuntime(op) __check_cuda_runtime((op), #op, __FILE__, __LINE__)
|
|
|
|
bool __check_cuda_runtime(cudaError_t code, const char* op, const char* file, int line);
|
|
|
|
#define BLOCK_SIZE 8
|
|
|
|
//note: resize rgb with padding
|
|
void resizeDevice(const int& batch_size, float* src, int src_width, int src_height,
|
|
float* dst, int dstWidth, int dstHeight,
|
|
float paddingValue, trtUtils::AffineMat matrix);
|
|
|
|
//overload:resize rgb with padding, but src's type is uin8
|
|
void resizeDevice(const int& batch_size, unsigned char* src, int src_width, int src_height,
|
|
float* dst, int dstWidth, int dstHeight,
|
|
float paddingValue, trtUtils::AffineMat matrix);
|
|
|
|
// overload: resize rgb/gray without padding
|
|
void resizeDevice(const int& batchSize, float* src, int srcWidth, int srcHeight,
|
|
float* dst, int dstWidth, int dstHeight,
|
|
trtUtils::ColorMode mode, trtUtils::AffineMat matrix);
|
|
|
|
void bgr2rgbDevice(const int& batch_size, float* src, int srcWidth, int srcHeight,
|
|
float* dst, int dstWidth, int dstHeight);
|
|
|
|
void normDevice(const int& batch_size, float* src, int srcWidth, int srcHeight,
|
|
float* dst, int dstWidth, int dstHeight,
|
|
trtUtils::InitParameter norm_param);
|
|
|
|
void hwc2chwDevice(const int& batch_size, float* src, int srcWidth, int srcHeight,
|
|
float* dst, int dstWidth, int dstHeight);
|
|
|
|
void decodeDevice(trtUtils::InitParameter param, float* src, int srcWidth, int srcHeight, int srcLength, float* dst, int dstWidth, int dstHeight);
|
|
|
|
// nms fast
|
|
void nmsDeviceV1(trtUtils::InitParameter param, float* src, int srcWidth, int srcHeight, int srcArea);
|
|
|
|
// nms sort
|
|
void nmsDeviceV2(trtUtils::InitParameter param, float* src, int srcWidth, int srcHeight, int srcArea,
|
|
int* idx, float* conf); |