下载 GPU Mats 的矢量到主机

Download a vector of GpuMats to host

本文关键字:主机 GPU Mats 下载      更新时间:2023-10-16

我有一个 std::vector,我想在我的主机上访问它。我将如何执行一个批处理 memcpy 操作(或下载操作(以返回 cv::Mats 的向量?我在 Jetson TX2 上运行它,所以我有能力做统一的共享内存。

最简单的方法是:

std::vector<cv::Mat> host;
// pre-allocate the memory.
host.reserve(devices.size());
cv::Mat tmp;
// Download each image one by one.
for(cv::cuda::GpuMat& d : devices)
{
d.download(tmp);
// Remember that the assignation between two cv::Mat does not do a copy but a reference sharing.
host.push_back(tmp.clone());
}