暗通道去雾改进算法及实现

上次搞的暗通道去雾的算法交给老师就算是交差了,当时也就是个调研而已。前几天又被老师叫过去说还是需要720p(1280*720)图像的实时处理,看能不能再做一些优化,让我和一个职工商量着来,于是又看了两天的去雾。还是有一些进展,总结一下。

回顾

还是可以回顾一下暗通道去雾的流程,在这里,其中这个模型是最重要的:

在这个模型中,要获得的只有两个,一个是透射率图t(x),一个是大气光值A,大气光值的获得方法很多,包括暗通道去雾中找最亮的0.5%的最低值,或者是有一个四叉树寻找的方法:对比度暗通道去雾,这片文章可以看做是对何凯明暗通道去雾的一个扩展,对天空具有比较好的免疫性。作者说效果不错,好不容易调通给的代码发现效果一般,而且算法比暗通道去雾还复杂一些,再就没有细看。

透射率这里,何提出的导向滤波是不错的,效果也还可以,但是实测下来还是这部分花费了大部分时间,后来使用了先下采样然后导向滤波之后再插值的快速算法,速度上还是有瓶颈。

如果在这个去雾模型上要有所改进的话,就只能是对透射率图这里改进了。

改进

查资料的过程中发现一篇文章:一种可实时处理的图像去雾算法的实现,文中给出了算法的详细步骤:

但说实话这个图里面的公式写的不太严谨,比如5式中的Mav应该是要进行归一化的,其他的地方倒还容易理解,特别是5式这个透射率的求法简直不要太简洁啊,看了博客中作者给出的效果图以及后面作者给的程序进行测试,发现效果还是不错的,而且从算法复杂度上来看,是要比原始的暗通道去雾有些优势的,最起码没有麻烦的导向滤波(导向滤波里有大量的求均值和滤波操作)。

然后就开始写代码,先用python写了一个版本,很快就写完了,测试下来是要比暗通道去雾快一些,按照C++版本应该比python版本提高近一倍的效率来说,是差不多可以到实时了,于是就动手写了。

这还是上周四的事情,周四晚上打球把脚崴了,周五肿了很大,自然是下不了楼就窝在宿舍里写程序,虽然说OpenCv已经提供了大量的API,写起来还是没有python顺手,主要是数据类型的转换。
比如说如果要做除法的话肯定是要把数据转换成float的,不能用uchar来做,因为当时用的指针来写的min_BGR函数,如果要换乘float的话,这里面的指针也都得改,用uchar类型的指针来访问float的话肯定是要崩溃的。诸如此类的问题吧。

大概到晚上吃饭的时候就已经完全调通了,中间出了一个莫名其妙的问题卡了很久,贴在下面:

vector<Mat> Img_split_dehaze;
    //Mat div_tmp;

    for (int i = 0; i < 3; i++)
    {
        /*cv::divide(Fenzi[i], fenmu, div_tmp);
        cout<<div_tmp;
        Img_split_dehaze.push_back(div_tmp);*/        
        
        //一开始是注释的这里错了,从最前面一点一点检查才发现,不过不知道哪里有问题,改成new对象就可以了,神奇!

        Mat *tmp = new Mat;
        cv::divide(Fenzi[i], fenmu, *tmp);
        Img_split_dehaze.push_back(*tmp);
        delete tmp;
    }

我一开始使用的注释掉的那一部分来做这个除法,然后储存到vector<Mat>里,然后三次push_back进去的竟然是同一个东西,导致我最后merge的图像就是灰度图,一开始怎么着也没发现这里有错误(现在也不知道为什么错了,我在循环里cout出来分明是不一样的)。
后来对照着python里的结果一点点检查才发现时这里的错误,浪费了很长的时间,不知道怎么处理就用new一个对象来做,就没有问题了。
i5_7500cpu上大概可以跑打25fps了,笔记本上要慢2/5差不多。
算法的实现就差不多是这样了,遍历像素我还是用的at运算符来做,毕竟最合乎习惯,而且release模式下速度损失不大。

优化

图像算法层面的优化就没有什么了,基本上不存在可以优化的地方,但是还是有一些地方可以进行优化。主要做了有两点。

式5求min(Mat1,Mat2)

一开始使用了vector<Mat>把两个式子merge到一张二通道的图里,然后再调用函数MIN_BRG_32F()(自己写的)来做这个工作,我觉得vector的创建以及调用merge可能还是花费了一些时间,于是直接来做,于是变成了:

M_minfilter_32f = M_minfilter_32f*m;

    int rows = M_minfilter_32f.rows;
    int cols = M_minfilter_32f.cols;
    Mat res = Mat::zeros(Size(cols,rows), CV_32FC1);
    
    for (int i = 0; i < rows; i++)
    {
        for (int j = 0; j < cols; j++)
        {

            res.at<float>(i, j) = std::min(M_minfilter_32f.at<float>(i,j), M_min_32f.at<float>(i, j));
        }
    }

速度大概提升了10%吧。

式7用查表法来做

这个改进是实质性的,这也是受了那篇博客中作者想法的启发,用查表来做,可以用查表法来做的基础是:所有对应的操作都是点对点的操作,而且H和L都是[0-255]uchar类型,另外还有一个参数A,是一个独立的大气光值,也是0-255的,所以我们可以做一个二维的查找表,一维是H的取值,一维是L的取值。这样的话就不用多次调用除法,只需要根据A来建立一个256*256大小的查找表就可以了。
查找表的建立: 我直接使用vector来做,索引起来也非常容易。

vector<vector<uchar>> Function7_table(const float &A)      
{
    vector<vector<uchar>> Table = vector<vector<uchar>>(256, vector<uchar>(256, 0));
    //cout << Table.size() << endl;
    double Value=0;
    for (int Y = 0; Y < 256; Y++)
    {
        for (int X = 0; X < 256; X++)
        {
            Value = (Y - X) / (1 - X/A);       //function_7
            if (Value > 255)
                Value = 255;
            else if (Value < 0)
                Value = 0;
            
            Table[Y][X] = Value;
            //printf("%d\t", Table[Y][X]);  这是个大坑,cout输出uchar的时候是按照ascii码输出的,一开始还以为不对
        }
    }
    return Table;
}

索引的时候非常方便,对于一个H(x)和L(x)的元素来说,只需要按照对应的位置来查找这个表就可以了:

vector<vector<uchar>> Table;
    Table = Function7_table(A);
    
    Mat ImgDefog = Img.clone();

    int row_Num = ImgDark_32f.rows;
    int col_Num = ImgDark_32f.cols;
    //获得行列
    for (int i = 0; i < row_Num; i++)
    {
        for (int j = 0; j < col_Num; j++)
        {
            ImgDefog.at<Vec3b>(i, j)[0] = Table[Img.at<Vec3b>(i, j)[0]][int(Lx.at<float>(i, j))];
            ImgDefog.at<Vec3b>(i, j)[1] = Table[Img.at<Vec3b>(i, j)[1]][int(Lx.at<float>(i, j))];
            ImgDefog.at<Vec3b>(i, j)[2] = Table[Img.at<Vec3b>(i, j)[2]][int(Lx.at<float>(i, j))];
            //处理三个通道
        }
    }

经过测试,速度有40%左右的提升,这样的话实时应该是很轻松的就可以达到了,查表果然是个好东西。
附一张效果图:效果还可以。

代码:FastDefog

我的博客即将搬运同步至腾讯云+社区,邀请大家一同入驻:https://cloud.tencent.com/developer/support-plan?invite_code=25kceejka15ww

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,324评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,303评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,192评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,555评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,569评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,566评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,927评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,583评论 0 257
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,827评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,590评论 2 320
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,669评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,365评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,941评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,928评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,159评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,880评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,399评论 2 342

推荐阅读更多精彩内容