使用 Nvidia NPP 调整图像大小时出现未记录的调整大小错误

Jas*_*son 2 c++ cuda gpu image-processing npp

我正在尝试使用 Nvidia 性能基元库来调整图像大小,但该nppiResize_8u_C3R函数抛出NPP_RESIZE_FACTOR_ERROR的错误未在文档中列出作为该函数的错误返回代码之一。这是我希望简单的代码:

#include <iostream>
#include <nppi.h>

int image_a_pitch;
NppiSize image_a_size = {.width = 960, .height = 540};
NppiRect image_a_roi = {.x = 0, .y = 0, .width = 960, .height = 540};
Npp8u* image_a = nppiMalloc_8u_C3(960, 540, &image_a_pitch);

int image_b_pitch;
NppiSize image_b_size = {.width = 960, .height = 540};
NppiRect image_b_roi = {.x = 0, .y = 0, .width = 960, .height = 540};
Npp8u* image_b = nppiMalloc_8u_C3(960, 540, &image_b_pitch);

NppStatus result = nppiResize_8u_C3R(image_a, image_a_pitch, image_a_size, image_a_roi, image_b, image_b_pitch, image_b_size, image_b_roi, NPPI_INTER_SUPER);

if (result != NPP_SUCCESS) {
    std::cerr << "Error executing Resize -- code: " << result << std::endl;
}
Run Code Online (Sandbox Code Playgroud)

您可能会注意到我实际上并没有调整图像的大小,是的,这是事实,但我正在尝试为该错误创建最简单的情况。从我一直在测试的情况来看,我对图像使用什么尺寸似乎并不重要,我仍然得到相同的返回错误代码,这似乎不是该函数的有效错误代码,而是一个错误折旧代码nppiResizeSqrPixel_8u_C3R。

我非常感谢一些帮助指出我在设置导致此错误的库调用时所做的任何错误。

Rob*_*lla 5

查看 jpegNPP CUDA 示例代码,我观察到以下序列:

    NppiInterpolationMode eInterploationMode = NPPI_INTER_SUPER;

    if (nScaleFactor >= 1.f)
        eInterploationMode = NPPI_INTER_LANCZOS;

    NPP_CHECK_NPP(nppiResize_8u_C1R(..., eInterploationMode));
}
Run Code Online (Sandbox Code Playgroud)

NPPI_INTER_SUPER当我在示例中将插值模式从 切换为 时NPPI_INTER_LANCZOS,错误就消失了(CUDA 9.1,linux)。我怀疑 NPP 库对此函数有一些未记录的(AFAICT)要求,即NPPI_INTER_SUPER当调整大小比例因子为 1.0 或更大(您的是 1.0,它是 sourceROI/destROI 的比率)时,不使用插值模式。

我建议您研究该示例代码中的用法,并在您的代码中放置类似的机制。

我已经提出请求,要求使用适当的信息更新 CUDA 文档。我没有任何进一步的细节。欢迎您在http://developer.nvidia.com上提交您自己的错误

更新:我已与 CUDA 团队确认,计划在未来的 CUDA 版本中对此进行文档更新。简而言之,使用的要求NPPI_INTER_SUPER是必须进行缩小。也就是说,X 源 ROI 尺寸除以 X 目标 ROI 尺寸的比率必须大于 1.0。同样,Y 源 ROI 尺寸除以 Y 目标 ROI 尺寸的比率必须大于 1.0。如果不满足其中任何一个条件,则NPPI_INTER_SUPER无法使用插补模式。例如,NPPI_INTER_LANCZOS可以使用 模式来代替。