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

[CVCUDA] Update CV-CUDA to v0.2.1, add vision processor C++ tutorial (#1678)

Browse Source 
* update cvcuda 0.2.0 -> 0.2.1

* add cpp tutorials demo

* fix reviewed problem
This commit is contained in:
guxukai
2023-03-24 16:57:35 +08:00
committed by GitHub
parent f0235a4c4e
commit 631f94d276
17 changed files with 170 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fastdeploy/vision/common/processors/cvcuda_utils.cc
View File

@@ -43,8 +43,8 @@ nvcv::ImageFormat CreateCvCudaImageFormat(FDDataType type, int channel,
return nvcv::FMT_BGRf32;
}
nvcv::TensorWrapData CreateCvCudaTensorWrapData(const FDTensor& tensor,
Layout layout) {
std::shared_ptr<nvcv::TensorWrapData> CreateCvCudaTensorWrapData(
const FDTensor& tensor, Layout layout) {
FDASSERT(tensor.shape.size() == 3,
"When create CVCUDA tensor from FD tensor,"
"tensor shape should be 3-Dim,");
@@ -76,7 +76,7 @@ nvcv::TensorWrapData CreateCvCudaTensorWrapData(const FDTensor& tensor,
nvcv::TensorDataStridedCuda tensor_data(
nvcv::TensorShape{req.shape, req.rank, req.layout},
nvcv::DataType{req.dtype}, buf);
return nvcv::TensorWrapData(tensor_data);
return std::make_shared<nvcv::TensorWrapData>(tensor_data, nullptr);
}
void* GetCvCudaTensorDataPtr(const nvcv::TensorWrapData& tensor) {