This statement was keeping HAAR cascades from leveraging opencl on nvidia devices. "localSize" on the featureEvaluator remains Size(0, 0) which sets the bool "use_ocl" to false. Adding this allows NVidia GPUs to leverage opencl HAAR Cascades

This commit is contained in:
Geoff McIver 2017-10-11 09:32:38 +13:00
parent af8ed9d09f
commit b2d8e8c508

View File

@ -609,7 +609,7 @@ bool HaarEvaluator::read(const FileNode& node, Size _origWinSize)
localSize = lbufSize = Size(0, 0);
if (ocl::haveOpenCL())
{
if (ocl::Device::getDefault().isAMD() || ocl::Device::getDefault().isIntel())
if (ocl::Device::getDefault().isAMD() || ocl::Device::getDefault().isIntel() || ocl::Device::getDefault().isNVidia())
{
localSize = Size(8, 8);
lbufSize = Size(origWinSize.width + localSize.width,