用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)

获取更多代码请私信~

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐