高斯卷积时边界处的急剧过渡(2d)

时间:2015-05-22 18:13:46

标签: c++ gaussian convolution

我正在尝试使用高斯2D卷积来模糊矩阵。 但是我在边界元素处得到了明显的过渡。

这是我正在运行的一段代码:

// create 1D Kernel
void createGaussianKerenel_1D() {
    unsigned kernelSize = 2 * kernelRad_ + 1;
    gaussian1Dkernel_ = vector<double>(kernelSize);
    double sigma = (double)kernelRad_;

    double sum = 0.0;
    for(unsigned i = 0; i < kernelSize; ++i) {
        gaussian1Dkernel_[i] = gaussian(i, sigma);
        sum += gaussian1Dkernel_[i];
    }

    // normalize
    for(unsigned i = 0; i < kernelSize; ++i) {
        gaussian1Dkernel_[i] /= sum;
        cout << gaussian1Dkernel_[i] << endl;
    }
}

// gaussian function
double gaussian(unsigned int i, double sigma) const {
    double x = ((double)i - (double)kernelRad_) / sigma;

    return exp(-x * x / 2);
}

// do Separable 2D Convolution (in place)
// my initialMatrix_ is of yn_ x xn_ size
void getBlurredThermalMap() {
    assert(!gaussian1Dkernel_.empty());
    vector<vector<double> > tmpMatrix(yn_);
    unsigned kernelSize = 2 * kernelRad_ + 1;

    // in x direction
    for(unsigned i = 0; i < yn_; ++i) {
        for(unsigned j = 0; j < xn_; ++j) {
            double approxVal = 0.0;
            for(unsigned row = 0; row < kernelSize; ++row) {
                unsigned neighbor_j = j + row - kernelRad_;
                // ignore values that are out of bound
                if(neighbor_j >= 0 && neighbor_j < xn_) {
                    approxVal += initialMatrix_[i][neighbor_j] * gaussian1Dkernel_[row];
                }
            }
            tmpMatrix[i].push_back(approxVal);
        }
    }

    // in y direction
    for(unsigned j = 0; j < xn_; ++j) {
        for(unsigned i = 0; i < yn_; ++i) {
            double approxVal = 0.0;
            for(unsigned col = 0; col < kernelSize; ++col) {
                unsigned neighbor_i = i + col - kernelRad_;
                if(neighbor_i >= 0 && neighbor_i < yn_) {
                    approxVal += tmpMatrix[neighbor_i][j] * gaussian1Dkernel_[col];
                }
            }
            initialMatrix_[i][j] = approxVal;
        }
    }
}

即,我正在为边界元素使用相同的内核。 我已经在100x100矩阵和2个半径的内核上测试了这个代码。 而且,例如,我在1,97和2,97之间的元素之间存在很大的差异,尽管在这些位置的初始矩阵中没有明显的过渡。

在计算边界元素的近似值时,我可能需要更改内核吗?

提前致谢。

2 个答案:

答案 0 :(得分:1)

这可能是因为你没有正确处理边界条件。在你的测试中:

if(neighbor_i >= 0 && neighbor_i < yn_)

第一部分将永远为真,因为neighbor_iunsigned,因此总是正数。您可能希望将其更改为有符号值,稍微修改其声明。您的编译器可以使用适当的警告标志检查这些错误(尝试-Wall -Wextra)。

修改 实际上,测试可能不是您问题的原因,因为您使用相对较小的图片,使neighbor_i的值大于yn_当试图在其中存储负值时。

另外,请使用库来进行卷积。特别是有相当好的和有效的近似值(Canny-Deriche,傅里叶域中的产品,......)到高斯模糊。

答案 1 :(得分:0)

我解决了这个问题如下:

不要在createGaussianKerenel_1D()函数中规范化内核。 相反,在getBlurredThermalMap()函数中执行如下操作:

void getBlurredThermalMap() {
    assert(!gaussian1Dkernel_.empty());
    vector<vector<double> > tmpMatrix(yn_);
    unsigned kernelSize = 2 * kernelRad_ + 1;

    // in x direction
    for(unsigned i = 0; i < yn_; ++i) {
        for(unsigned j = 0; j < xn_; ++j) {
            double approxVal = 0.0;
            double sumNorm = 0.0;
            for(unsigned row = 0; row < kernelSize; ++row) {
                unsigned neighbor_j = j + row - kernelRad_;
                // ignore values that are out of bound
                if(neighbor_j >= 0 && neighbor_j < xn_) {
                    approxVal += initialMatrix_[i][neighbor_j] * gaussian1Dkernel_[row];
                    sumNorm += gaussian1Dkernel_[row];
                }
            }
            approxVal /= sumNorm;
            tmpMatrix[i].push_back(approxVal);
        }
    }

    // in y direction
    for(unsigned j = 0; j < xn_; ++j) {
        for(unsigned i = 0; i < yn_; ++i) {
            double approxVal = 0.0;
            double sumNorm = 0.0;
            for(unsigned col = 0; col < kernelSize; ++col) {
                unsigned neighbor_i = i + col - kernelRad_;
                if(neighbor_i >= 0 && neighbor_i < yn_) {
                    approxVal += tmpMatrix[neighbor_i][j] * gaussian1Dkernel_[col];
                    sumNorm += gaussian1Dkernel_[row];
                }
            }
            approxVal /= sumNorm;
            initialMatrix_[i][j] = approxVal;
        }
    }
}