apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-21 15:49:31 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

[CVCUDA] PaddleDetection preprocessor support CV-CUDA (#1493)

Browse Source 
* ppdet preproc use manager

* pad_to_size chw opencv

* pad_to_size chw flycv

* fix pad_to_size flycv

* add warning message

* cvcuda convert cubic to linear, padToSize cvcuda

* stridedpad cvcuda

* fix flycv include

* fix flycv include

* fix flycv build

* cast cvcuda

* fix pybind

* fix normalize permute cuda

* base processor move funcs to cc

* Update pad_to_size.cc
This commit is contained in:
Wang Xinyu
2023-03-10 12:43:57 +08:00
committed by GitHub
parent 9ee2118e1f
commit cb7c8a07d4
23 changed files with 537 additions and 239 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/vision/common/processors/cvcuda_utils.h
View File

@@ -18,8 +18,9 @@
#include "fastdeploy/vision/common/processors/mat.h"
#ifdef ENABLE_CVCUDA
#include "nvcv/Tensor.hpp"
#include <nvcv/Tensor.hpp>
#include <nvcv/ImageBatch.hpp>
#include <cvcuda/Types.h>
namespace fastdeploy {
namespace vision {
@@ -32,6 +33,7 @@ void* GetCvCudaTensorDataPtr(const nvcv::TensorWrapData& tensor);
nvcv::ImageWrapData CreateImageWrapData(const FDTensor& tensor);
void CreateCvCudaImageBatchVarShape(std::vector<FDTensor*>& tensors,
nvcv::ImageBatchVarShape& img_batch);
NVCVInterpolationType CreateCvCudaInterp(int interp);
} // namespace vision
} // namespace fastdeploy