Method formatToCvMat

src/openpose/core/opOutputToCvMat.cpp:77–130

    }

    Matrix OpOutputToCvMat::formatToCvMat(const Array<float>& outputData)
    {
        try
        {
            // Sanity check
            if (outputData.empty())
                error("Wrong input element (empty outputData).", __LINE__, __FUNCTION__, __FILE__);
            // Final result
            cv::Mat cvMat;
            // CPU version
            if (!mGpuResize)
            {
                // outputData to cvMat
                // Equivalent: outputData.getConstCvMat().convertTo(cvMat, CV_8UC3);
                const cv::Mat constCvMat = OP_OP2CVCONSTMAT(outputData.getConstCvMat());
                constCvMat.convertTo(cvMat, CV_8UC3);
            }
            // CUDA version
            else
            {
                #ifdef USE_CUDA
                    // (Free and re-)Allocate temporary memory
                    if (mOutputMaxSizeUChar < *spOutputMaxSize)
                    {
                        mOutputMaxSizeUChar = *spOutputMaxSize;
                        cudaFree(pOutputImageUCharCuda);
                        cudaMalloc((void**)&pOutputImageUCharCuda, sizeof(unsigned char) * mOutputMaxSizeUChar);
                    }
                    // Float ptr --> unsigned char ptr
                    const auto volume = (int)outputData.getVolume();
                    uCharImageCast(pOutputImageUCharCuda, *spOutputImageFloatCuda, volume);
                    // Allocate cvMat
                    cvMat = cv::Mat(outputData.getSize(0), outputData.getSize(1), CV_8UC3);
                    // CUDA --> CPU: Copy output image back to CPU
                    cudaMemcpy(
                        cvMat.data, pOutputImageUCharCuda, sizeof(unsigned char) * volume,
                        cudaMemcpyDeviceToHost);
                    // Indicate memory was copied out
                    *spGpuMemoryAllocated = false;
                #else
                    error("You need to compile OpenPose with CUDA support in order to use GPU resize.",
                        __LINE__, __FUNCTION__, __FILE__);
                #endif
            }
            // Return cvMat
            const Matrix opMat = OP_CV2OPMAT(cvMat);
            return opMat;
        }
        catch (const std::exception& e)
        {
            error(e.what(), __LINE__, __FUNCTION__, __FILE__);
            return Matrix();
        }
    }
}