Как получить доступ к буферу OpenCV UMat (gpu) из OpenCL? - PullRequest
5 голосов
/ 17 октября 2019

Введение:

Цель - захватить видео с помощью OpenCV и использовать его в качестве входных данных для программы OpenCL . Передача обоих должна быть как можно более эффективной (если это не будет проблемой, зачем использовать OpenCL, верно?).
Я читал, что OpenCV использует OpenCL внутри (UMat),и что я мог получить доступ к буферу GPU, открыв UMat::handle. Однако мои попытки сделать это до сих пор не увенчались успехом.

Намерение состоит в том, чтобы повторно использовать буфер UMat в качестве входных данных для OpenCL kernels и, в конечном итоге, получить результат в виде изображения обратно к другому. UMat для его отображения.

Каркас OpenCV предназначен только для создания входных данных для программы, поэтому я не заинтересован в использовании оболочки OpenCV CL (cv::ocl), а скорее использую обычный OpenCL (* * тысяча двадцать-одиной). Это позволяет избежать включения / связывания инфраструктуры OpenCV в полное программное обеспечение.

Вопрос:

Как получить доступ к буферу OpenCV UMat через OpenCL?

  • Использование буфера UMat в качестве буфера OpenCL (первая опция)
  • Перемещение буфера UMat в буфер OpenCL в графическом процессоре. (второй вариант)

Чего я еще достиг:

  • OpenCL работают совершенно автономно
  • OpenCV работают совершенно автономно
  • Преобразование дескриптора UMat :: в компиляцию cl :: Buffer
  • Указанный буфер недопустим.

ОТКАЗ ОТ ОТВЕТСТВЕННОСТИ: пожалуйста, будьте добры,так как это всего лишь минимальный пример для этого вопроса.

#include <iostream>
#include <vector>
#include <cassert>

#define __CL_ENABLE_EXCEPTIONS // enable exceptions instead of error-codes
#define CL_TARGET_OPENCL_VERSION 120
#include <CL/cl.hpp>
#include <opencv2/opencv.hpp>
#include <opencv2/core/ocl.hpp>

using namespace cv;
using namespace std;

int main()
{
    // OPENCL STUFF
    // Very simplified/basic/stupid/naive OpenCL context creation
    std::vector<cl::Platform> platforms;
    cl::Platform::get(&platforms);
    assert(platforms.size()>0);
    std::vector<cl::Device> devices;
    platforms[0].getDevices( CL_DEVICE_TYPE_ALL, &devices);
    assert(devices.size()>0);
    cl_context_properties prop[3] =
    {
        CL_CONTEXT_PLATFORM,
        (cl_context_properties)(platforms[0])(),
        0
    };
    cl::Context context( devices[0], prop, nullptr, nullptr);

    std::string kernelStr = R"DELIMITER(
    kernel void replaceRB( global uchar3* content)
    {
        const size_t globalId = get_global_id(0);

        private uchar3 byte = content[globalId];
        char aux = byte.z;
        byte.z = byte.x;
        byte.x = aux;
        content[globalId] = byte;
    }
    )DELIMITER";

    cl::Program::Sources sources;
    sources.push_back(std::make_pair<const char*, size_t>(kernelStr.data(), kernelStr.size()));
    cl::Program program(context, sources);

    try
    {
        program.build({devices[0]}, "");
    }
    catch (...)
    {
        std::cout << program.getBuildInfo<CL_PROGRAM_BUILD_LOG>(devices[0]) << std::endl;
    }

    std::vector<cl::Kernel> kernels;
    program.createKernels(&kernels);
    assert(kernels.size()>0);

    cl::CommandQueue queue(context, devices[0]);

    // OPENCV STUFF
    ocl::setUseOpenCL(true);
    cv::ocl::attachContext(platforms[0].getInfo<CL_PLATFORM_NAME>(), platforms[0](), context(), devices[0]());
    assert(ocl::haveOpenCL());

    cout << cv::ocl::Context::getDefault().ndevices() << " GPU devices are detected." << endl;

    VideoCapture cap(0); //Camera
    //VideoCapture cap("SampleVideo_1280x720_1mb.mp4"); //Video example
    assert(cap.isOpened());

    UMat frame;
    assert(cap.read(frame));

    //MIX OF BOTH opencl and opencv
    //cl::Buffer buf(context,CL_MEM_READ_WRITE, 256); // This works
    cl::Buffer buf(*((cl_mem*)frame.handle(CL_MEM_READ_WRITE)));

    int result = kernels[0].setArg(0, buf);
    std::cout << result << " == " << CL_INVALID_MEM_OBJECT << std::endl;

    queue.enqueueNDRangeKernel(kernels[0], cl::NullRange, cl::NDRange(16), cl::NDRange(4));
    queue.flush();

    //DISPLAY RESULT?
    string window_name = "Test OpenCV and OpenCL";
    namedWindow(window_name);
    imshow(window_name, frame);
    waitKey(5000);

    return 0;
}
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...