即插即用的opencv去雾代码实现
何凯明去雾算法,python即插即用
·

用python实现了何凯明论文中的去雾算法,只用到了QT作为可视化和opencv。


def dehaze(self, image):
"""核心去雾函数"""
if image is None or image.ndim != 3 or image.shape[2] != 3:
raise ValueError("输入需要是3通道BGR图像")
# 转换为浮点数处理
image = image.astype(np.float32) / 255.0
# 1. 暗通道计算
dark_channel = self.get_dark_channel(image)
# 2. 大气光估计
atmosphere = self.estimate_atmosphere(image, dark_channel)
# 3. 透射率估计
transmission = self.estimate_transmission(image, atmosphere)
# 4. 导向滤波优化
refined_trans = self.guided_filter(image, transmission)
# 5. 图像恢复
result = self.recover_image(image, refined_trans, atmosphere)
return (result * 255).astype(np.uint8)
获取更多代码请私信~
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)