使用 opencv 在 GPU 上调整图像大小会产生空输出矩阵

Posted

技术标签:

【中文标题】使用 opencv 在 GPU 上调整图像大小会产生空输出矩阵【英文标题】:Resizing image on GPU with opencv produces empty output matrix 【发布时间】:2020-03-04 11:09:00 【问题描述】:

我正在运行这个简单的应用程序来在 GTX1080ti GPU 上调整图像大小:

#include <opencv2/opencv.hpp>
#include "opencv2/cudaimgproc.hpp"
#include "opencv2/cudawarping.hpp"
using namespace std;
using namespace cv;
using namespace cv::cuda;
static void gpuResize(Mat in, Mat out)
    double k = in.cols/416.;
    cuda::GpuMat gpuInImage;
    cuda::GpuMat gpuOutImage;
    gpuInImage.upload(in);
    const Size2i &newSize = Size(416, in.rows / k);
    cout << "newSize " << newSize<< endl;
    cuda::resize(gpuInImage, gpuOutImage, newSize);
    gpuOutImage.download(out);


int main()
    Mat im = Mat::zeros(Size(832,832),CV_8UC3);
    Mat out;
    if (getCudaEnabledDeviceCount() == 0)
        return cerr << "No GPU found or the library is compiled without CUDA support" << endl, -1;
    
    cv::cuda::printShortCudaDeviceInfo(cv::cuda::getDevice());
    gpuResize(im,out);
    cout << "real size="<<out.size() <<endl;

我得到的输出:

Device 0:  "GeForce GTX 1080 Ti"  11177Mb, sm_61, Driver/Runtime ver.10.10/10.10
newSize [416 x 416]
real size=[0 x 0]

为什么输出矩阵的大小为 0x0,我应该如何解决?

【问题讨论】:

您应该传递out by reference 而不是按值传递吗? 【参考方案1】:

这看起来像是您如何解析函数的问题。该函数将修改其本地副本,并且一旦您的函数返回。原始输出没有被修改。尝试通过引用解析。

【讨论】:

这是根本原因,亲爱的约书亚。但我有点困惑,我认为 Mat 打算作为共享指针工作?当我按值传递 Mat 时,不会复制实际像素,对吗?您能否在答案中澄清这一点?

以上是关于使用 opencv 在 GPU 上调整图像大小会产生空输出矩阵的主要内容,如果未能解决你的问题,请参考以下文章

使用 OpenCV 调整图像大小

为啥在 OpenCv 中调整图像大小会降低相机校准的重投影误差?

如何在 OpenCV 中将图像调整为特定大小?

如何在opencv python中调整PNG图像的大小? [复制]

当使用 OpenCV 完成图像加载和调整大小时,Resnet50 会产生不同的预测

GPU 上的图像调整大小比 cv::resize 慢