我正在尝试使用高斯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之间的元素之间存在很大的差异,尽管在这些位置的初始矩阵中没有明显的过渡。
在计算边界元素的近似值时,我可能需要更改内核吗?
提前致谢。
答案 0 :(得分:1)
这可能是因为你没有正确处理边界条件。在你的测试中:
if(neighbor_i >= 0 && neighbor_i < yn_)
第一部分将永远为真,因为neighbor_i
是unsigned
,因此总是正数。您可能希望将其更改为有符号值,稍微修改其声明。您的编译器可以使用适当的警告标志检查这些错误(尝试-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;
}
}
}