在好例子网,分享、交流、成长!
您当前所在位置:首页C/C++ 开发实例图形和图像处理 → c++ 图像去雾算法 示例源码

c++ 图像去雾算法 示例源码

图形和图像处理

下载此实例
  • 开发语言:C/C++
  • 实例大小:0.86M
  • 下载次数:54
  • 浏览次数:628
  • 发布时间:2017-12-23
  • 实例类别:图形和图像处理
  • 发 布 人:wuzhijia2017
  • 文件格式:.rar
  • 所需积分:2
 相关标签: 去雾算法 图像增强

实例介绍

【实例简介】图像去雾算法,图像增强
【实例截图】

【核心代码】

#include "hazeremovehandler.h"

HazeRemoveHandler::HazeRemoveHandler()
{
    kernel_size_ = 15;
    bright_percent_ = 0.1;
    A_[0] = A_[1] = A_[2] = 0;
    A_average =0;
    w_ = 0.95;
    r_ = 120;
    eps_ = 0.001*0.001;
    t0_ = 0.1;
}


// This Method is to read original image and convert it to float data type and been normilized to 0~1.
// Input : input Mat type image.

void HazeRemoveHandler::ReadOriImage(Mat img)
{
    ori_image_ = Mat(img.rows,img.cols,CV_32FC3);
    img.convertTo(ori_image_,CV_32FC3);

    // Normilize to 0~1
    ori_image_ = ori_image_/255;
}

void HazeRemoveHandler::ShowImage(int type)
{
    switch (type) {
    case 0:
        //
        // imshow function notice
        // for different data types
        // 1. 8-bit unsigned char : show directly
        // 2. 16-bit unsigned or 32-bit integer : pixel value/256, [0,255*256] -> [0,255]
        // 3. 32-bit floating-point : pixel value*255, [0,1]->[0,255]
        //
        imshow("Original Image",ori_image_);
        break;
    case 1:
        imshow("Darkchannel Image",dark_channel_image_);
        break;
    case 2:
        imshow("Transmission Map",transmission_image_);
        break;
    case 3:
        imshow("Smoothed Transmission Map",transmission_smoothed_image_);
        break;
    case 4:
        imshow("Haze Free Image",haze_free_image_);
        break;
    default:
        break;
    }
}

// This Method is to get the dark channel of the original image
// //J^{dark}(x)=min( min( J^c(y) ) )
// Output :
//          -1: means it has not load original image, should do ReadOriImage before.
//          0: means computing dark channel succeed.

int HazeRemoveHandler::DarkChannelPrior()
{
    if( (ori_image_.cols==0) || (ori_image_.rows==0))
    {
        return -1;
    }

    // Compute min value for every pixel during different three channels(R,G,B)
    Mat dark_tmp = Mat::zeros(ori_image_.rows,ori_image_.cols,CV_32FC1);

    int rows = dark_tmp.rows;
    int cols = dark_tmp.cols;

    for(int i = 0; i < rows; i  )
        for(int j = 0; j < cols; j  )
        {
            dark_tmp.at<float>(i,j) = min(
                        min(ori_image_.at<Vec3f>(i,j)[0],ori_image_.at<Vec3f>(i,j)[1]),
                        min(ori_image_.at<Vec3f>(i,j)[0],ori_image_.at<Vec3f>(i,j)[2])
                        );
        }

    // Minfilter implemented by erode function
    erode(dark_tmp,dark_channel_image_,Mat::ones(kernel_size_,kernel_size_,CV_32FC1));

    return 0;

}
// This Method is to estimate the transmission map of the original image
// t(x) = 1- min( min( I^c(y)/A^c ) )
// Output :
//          -1: means it has not load original image or estimate the Atmospheric Light
//              , should do that before.
//          0: means estimate the transmission map succeed.

int HazeRemoveHandler::EstimateTransmission()
{
    A_average = (A_[0] A_[1] A_[2])/3;
    if(A_average>0.86)
        A_average = 0.86;
    if( (ori_image_.cols==0) || (ori_image_.rows==0) || (A_average ==0))
    {
        return -1;
    }
    Mat dark_tmp = Mat::zeros(ori_image_.rows,ori_image_.cols,CV_32FC1);

    int rows = dark_tmp.rows;
    int cols = dark_tmp.cols;

    for(int i = 0; i < rows; i  )
        for(int j = 0; j < cols; j  )
        {
            dark_tmp.at<float>(i,j) = min(
//                        min(ori_image_.at<Vec3f>(i,j)[0]/A_average,ori_image_.at<Vec3f>(i,j)[1]/A_average),
//                        min(ori_image_.at<Vec3f>(i,j)[0]/A_average,ori_image_.at<Vec3f>(i,j)[2]/A_average)
                        min(ori_image_.at<Vec3f>(i,j)[0],ori_image_.at<Vec3f>(i,j)[1]),
                        min(ori_image_.at<Vec3f>(i,j)[0],ori_image_.at<Vec3f>(i,j)[2])
                        );
        }
    Mat dark_hazeimage;
    // Minfilter implemented by erode function
    erode(dark_tmp,dark_hazeimage,Mat::ones(kernel_size_,kernel_size_,CV_32FC1));

    transmission_image_ = Mat(ori_image_.rows,ori_image_.cols,CV_32FC1);
    for(int i = 0; i < rows; i  )
        for(int j = 0; j < cols; j  )
        {
            transmission_image_.at<float>(i,j) = 1-w_*dark_hazeimage.at<float>(i,j);
        }
    return 0;
}

// This Method is to estimating the Atmospheric Light of the scene
// Output :
//          -1: means it has not compute dark channel image, should do DarkChannelPrior before.
//          0: means computing estimate succeed.

int HazeRemoveHandler::EstimateAtmoLight()
{
    if((dark_channel_image_.cols ==0 )||(dark_channel_image_.rows==0))
    {
        return -1;
    }

    int bripix_num;
    bripix_num = (int)dark_channel_image_.cols*dark_channel_image_.rows*bright_percent_/100;

    // Change rows*cols data -> 1*(rows*cols) data
    Mat dark_reshape = dark_channel_image_.reshape(0,1).clone();
    Mat sorted;
    sortIdx(dark_reshape,sorted,CV_SORT_EVERY_ROW CV_SORT_DESCENDING);
    vector<int> bripixel_idx;
    for(int i = 0 ; i < bripix_num; i  )
    {
        int pixel_rows;
        int pixel_cols;
        if(sorted.at<int>(0,i)%dark_channel_image_.cols==0)
        {
            pixel_rows = (sorted.at<int>(0,i) 1)/dark_channel_image_.cols-1;
            pixel_cols = (sorted.at<int>(0,i) 1)%dark_channel_image_.cols dark_channel_image_.cols;
        }
        else
        {
            pixel_rows = (sorted.at<int>(0,i) 1)/dark_channel_image_.cols;
            pixel_cols = (sorted.at<int>(0,i) 1)%dark_channel_image_.cols-1;
        }
        A_[0]  = ori_image_.at<Vec3f>(pixel_rows,pixel_cols)[0];
        A_[1]  = ori_image_.at<Vec3f>(pixel_rows,pixel_cols)[1];
        A_[2]  = ori_image_.at<Vec3f>(pixel_rows,pixel_cols)[2];
    }
    A_[0] /= bripix_num;
    A_[1] /= bripix_num;
    A_[2] /= bripix_num;
    return 0;
}

// This Method uses guided image filter to smooth the transimission map(keep edge and remove noise)
// Output :
//          -1: means it has not compute transimission map, should do that before.
//          0: means smooth succeed.
int HazeRemoveHandler::SmoothTransmissionMap()
{

    if((transmission_image_.cols ==0) || (transmission_image_.rows==0))
    {
        return -1;
    }

    transmission_smoothed_image_ = guidedFilter(ori_image_,transmission_image_,r_,eps_);

    return 0;
}

int HazeRemoveHandler::HazeFree()
{
    if((transmission_smoothed_image_.cols ==0) || (transmission_smoothed_image_.rows==0))
    {
        return -1;
    }

    haze_free_image_ = Mat(ori_image_.rows,ori_image_.cols,CV_32FC3);

    int rows = haze_free_image_.rows;
    int cols = haze_free_image_.cols;

    for(int i = 0; i < rows; i  )
        for(int j = 0; j < cols; j  )
        {
            float t_max;
            if(transmission_smoothed_image_.at<float>(i,j) > t0_)
                t_max = transmission_smoothed_image_.at<float>(i,j);
            else
                t_max = t0_;
            haze_free_image_.at<Vec3f>(i,j)[0] = (ori_image_.at<Vec3f>(i,j)[0]-A_average)/t_max   A_average;
            haze_free_image_.at<Vec3f>(i,j)[1] = (ori_image_.at<Vec3f>(i,j)[1]-A_average)/t_max   A_average;
            haze_free_image_.at<Vec3f>(i,j)[2] = (ori_image_.at<Vec3f>(i,j)[2]-A_average)/t_max   A_average;
        }
    return 0;
}

实例下载地址

c++ 图像去雾算法 示例源码

不能下载?内容有错? 点击这里报错 + 投诉 + 提问

好例子网口号:伸出你的我的手 — 分享

网友评论

发表评论

(您的评论需要经过审核才能显示)

查看所有0条评论>>

小贴士

感谢您为本站写下的评论,您的评论对其它用户来说具有重要的参考价值,所以请认真填写。

  • 类似“顶”、“沙发”之类没有营养的文字,对勤劳贡献的楼主来说是令人沮丧的反馈信息。
  • 相信您也不想看到一排文字/表情墙,所以请不要反馈意义不大的重复字符,也请尽量不要纯表情的回复。
  • 提问之前请再仔细看一遍楼主的说明,或许是您遗漏了。
  • 请勿到处挖坑绊人、招贴广告。既占空间让人厌烦,又没人会搭理,于人于己都无利。

关于好例子网

本站旨在为广大IT学习爱好者提供一个非营利性互相学习交流分享平台。本站所有资源都可以被免费获取学习研究。本站资源来自网友分享,对搜索内容的合法性不具有预见性、识别性、控制性,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,平台无法对用户传输的作品、信息、内容的权属或合法性、安全性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论平台是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二与二十三条之规定,若资源存在侵权或相关问题请联系本站客服人员,点此联系我们。关于更多版权及免责申明参见 版权及免责申明

;
报警