Opencv实现暗通道先验去雾算法

今天读了何凯明博士的《Single Image Haze Removal Using Dark Channel Prior》,用opencv实现了一遍。
其中,暗通道及T(x)采用腐蚀erode算法。
具体代码如下:

class hazeOutProcesser
{
    public:

    hazeOutProcesser(const Mat& srcImg)
    {
        this->srcImg = srcImg.clone();
    }

    hazeOutProcesser(){}

    void getDarkChannel(const Mat& src, Mat& dst)
    {
        Mat element = getStructuringElement(MORPH_RECT, Size(15, 15));
        // erode(src, dst, element);
        morphologyEx(src, dst, MORPH_ERODE, element);
        // Mat temp = dst.clone();
        // bilateralFilter(temp, dst, 25, 25*2, 25/2);
        cvtColor(dst, dst, COLOR_BGR2GRAY);
    }

    std::pair<Mat, Mat> getAtmosphere(const Mat& sortedIndex,
                         const Mat& srcImg,
                         double rate = 0.001)
    {
        long n_pixel = srcImg.rows * srcImg.cols;
        long n_search = n_pixel * rate;
        Mat srcImgVector = srcImg.reshape(1, n_pixel).clone(); //1channel, 3rows(b, g, r)
                                                       //(n_pixel, 3)
        cout << srcImgVector.rows << ", " << srcImgVector.cols << endl;
        Mat accumulator = Mat::zeros(1, 3, CV_32FC1); //[(32f)0, (32f)0, (32f)0]
        for(int i = 0; i < n_search; ++i) {
            int index = sortedIndex.at<int>(i);
            Mat t
  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值