导向滤波 Guided Filter 的 CUDA GPU版本

最新推荐文章于 2024-05-13 10:45:35 发布

文亘习丶

最新推荐文章于 2024-05-13 10:45:35 发布

阅读量481

点赞数 1

分类专栏：视频/图像处理和计算机视觉文章标签：图像处理 CUDA 计算机视觉

本文链接：https://blog.youkuaiyun.com/AgCl_LHY/article/details/131321377

版权

视频/图像处理和计算机视觉专栏收录该内容

2 篇文章

订阅专栏

该GitHub仓库提供了一个使用CUDA实现的GuidedFilter，能够在i9处理器和RTX4090GPU上快速处理1080P图像，平均时间约9.8ms。主要用于去雾应用，输入为灰度透射图，输出为浮点型。项目包含对RGB图像的灰度化处理，并提供了不同输入图像的处理时间数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Guided Filter Using CUDA

GitHub Repo : Plumess/Guided-Filter-Using-CUDA: A GPU version implementation of Guided Filter, using CUDA C/C++, calculates 1080P images in 10ms on 4090 (github.com)

这是导向滤波/引导滤波的一种GPU实现，经测试，在i9, RTX4090上运行，包括malloc和memcpy操作，1080P单帧处理可以达到9.8ms，可以直接或经过Resize缩放后加入到实时视觉任务中

介绍

测试场景为去雾工作，输入测试为RGB图的透射图，形式为灰度图，为了可视化保存为了数值范围为[0, 255]的png格式，在main函数中处理为[0, 1]的float类型；引导图为RGB原图，在main函数中处理为[0, 255]的unsigned char类型；输出为[0, 1]的float类型。

其中，RGB原图在进行Guided Filter之前按照BGR通道顺序，处理成了灰度图，如果有需要，可以自行修改guidedFilter.cu中的to_float_point函数。

实验结果

项目中给出的input测试结果如下

input/1_transmission.png
This Time: 140025 us
input/2_transmission.png
This Time: 9481 us
input/3_transmission.png
This Time: 10538 us
input/4_transmission.png
This Time: 9458 us

Average Time: 9825.67 us

PS：第一张处理包含GPU WarmUp过程，故计算平均时长时忽略了该值。