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); // 输入值 auchar1 b = make_uchar1(64); // 输入值 bfloat1 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