OpenCV CUDA模块设备层-----在GPU上计算两个uchar1类型像素值的反正切（arctangent）比值函数atan2()

• 操作系统：ubuntu22.04
• OpenCV版本：OpenCV4.9
• IDE:Visual Studio Code
• 编程语言：C++11

算法描述

对输入的两个 uchar1 像素值 a 和 b，先分别归一化到 [0.0, 1.0] 浮点区间，然后计算它们的 四象限反正切函数。

函数原型

__device__ __forceinline__ float1 cv::cudev::atan2 	( 	const uchar1 &  	a,
		const uchar1 &  	b 
	) 		

参数

• uchar1 CUDA 向量类型，表示一个单通道 8 位无符号整型（等价于 unsigned char 或 uint8_t）。

代码示例

#include <opencv2/cudev/functional.hpp>
#include <cstdio>

__global__ void test_atan2() {
    uchar1 a = make_uchar1(128); // 输入值 a
    uchar1 b = make_uchar1(64);  // 输入值 b
    float1 res = cv::cudev::atan2(a, b);
    printf("atan2(%d / 255, %d / 255) = %f radians\n", a.x, b.x, res.x);
}

int main() {
    test_atan2<<<1, 1>>>();
    cudaDeviceReset();
    return 0;
}

运行结果

atan2(128 / 255, 64 / 255) = 1.107149 radians