Необработанное исключение с cudaMemcpy2D

Question

Я новичок в C ++ (а также в Cuda и OpenCV), поэтому прошу прощения за любые ошибки на моей стороне. У меня есть существующий код, который использует Cuda. Недавно он работал с .png (который был декодирован) в качестве входных данных, но теперь я использую камеру для генерации живых изображений. Эти изображения являются новым входом для кода. Вот оно:

using namespace cv;

INT height = 2160;
INT width = 3840;
Mat image(height, width, CV_8UC3);
size_t pitch;
uint8_t* image_gpu;

// capture image
VideoCapture camera(0);
camera.set(CAP_PROP_FRAME_WIDTH, width);
camera.set(CAP_PROP_FRAME_HEIGHT, height);
camera.read(image);

// here I checked if image is definitly still a CV_8UC3 Mat with the initial height and width; and it is

cudaMallocPitch(&image_gpu, &pitch, width * 4, height);

// here I use cv::Mat::data to get the pointer to the data of the image:
cudaMemcpy2D(image_gpu, pitch, image.data, width*4, width*4, height, cudaMemcpyHostToDevice);

Код компилируется, но я получаю «Исключение» в последней строке (cudaMemcpy2D) со следующим кодом ошибки: Исключение, выброшенное в 0x00007FFE838D6660 (nvcuda.dll) в realtime.exe : 0xC0000005: Место чтения нарушения доступа 0x000001113AE10000.

Google не дал мне ответа, и я не знаю, что делать дальше.

Спасибо за любые подсказки!

Answer 1

Довольно общий c способ копирования OpenCV Mat в память устройства, выделенную с помощью cudaMallocPitch, заключается в использовании элемента step объекта Mat. Кроме того, при распределении памяти устройства необходимо помнить о том, как будет распределяться память устройства и как на нее будет скопирован объект Mat. Вот простой пример, демонстрирующий процедуру для видеокадра, снятого с использованием VideoCapture.

#include<iostream>
#include<cuda_runtime.h>
#include<opencv2/opencv.hpp>

using std::cout;
using std::endl;

size_t getPixelBytes(int type)
{
    switch(type)
    {
        case CV_8UC1:
        case CV_8UC3:
            return sizeof(uint8_t);
            break;
        case CV_16UC1:
        case CV_16UC3:
            return sizeof(uint16_t);
            break;
        case CV_32FC1:
        case CV_32FC3:
            return sizeof(float);
            break;
        case CV_64FC1:
        case CV_64FC3:
            return sizeof(double);
            break;
        default:
            return 0;
    }
}

int main()
{
    cv::VideoCapture cap(0);
    cv::Mat frame;

    if(cap.grab())
    {
        cap.retrieve(frame);
    }
    else
    {
        cout<<"Cannot read video"<<endl;
        return -1;
    }

    uint8_t* gpu_image;
    size_t gpu_pitch;

    //Get number of bytes occupied by a single pixel. Although VideoCapture mostly returns CV_8UC3 type frame thus pixelBytes is 1 , but just in case.
    size_t pixelBytes = getPixelBytes(frame.type());

    //Number of actual data bytes occupied by a row.
    size_t frameRowBytes = frame.cols * frame.channels * pixelBytes;

    //Allocate pitch linear memory on device
    cudaMallocPitch(&gpu_image, &gpu_pitch, frameRowBytes , frame.rows);

    //Copy memory from frame to device mempry
    cudaMemcpy2D(gpu_image, gpu_pitch, frame.ptr(), frame.step, frameRowBytes, frame.rows, cudaMemcpyHostToDevice);

   //Rest of the code ...
   return 0;
}

Отказ от ответственности: Код записывается в браузере. Еще не проверено. Пожалуйста, добавьте Проверка ошибок CUDA при необходимости