图像处理膨胀的C实现在结果图像上创建重复的图案

时间:2019-02-06 17:24:47

标签: c image-processing convolution

我正在尝试使用C实现膨胀操作。这是我已经完成的代码

#include <stdio.h>
#include <stdlib.h>
#include "image.h"

#define IMG_ROWS 1280
#define IMG_COLS 1024
#define KERNEL_DIM 5 //kernel size

unsigned char imgIn[IMG_ROWS][IMG_COLS]; //grayscale values array
unsigned char imgOut[IMG_ROWS][IMG_COLS];

//Kernel initialization
int kernel[KERNEL_DIM][KERNEL_DIM] = {
        {1,1,1,1,1},
        {1,1,1,1,1},
        {1,1,1,1,1},
        {1,1,1,1,1},
        {1,1,1,1,1}
};

int main()
{

    //fill the img matrix
    for(int idxRow = 0; idxRow < IMG_ROWS; idxRow++){
        for(int idxCol = 0; idxCol < IMG_COLS; idxCol++){
            imgIn[idxRow][idxCol] = img[idxRow*IMG_COLS + idxCol];
            imgOut[idxRow][idxCol] = 0;
        }
    }

    int max;
    int offset = KERNEL_DIM/2;
    //Iterates over the image ignoring the borders
    for(int idxRow = offset; idxRow < IMG_ROWS - offset; idxRow++){
        for(int idxCol = offset; idxCol < IMG_COLS - offset; idxCol++){
            max = 0;
            //Iterates over the kernel
            for(int krnRow = -offset; krnRow < offset + 1; krnRow++){
                for(int krnCol = -offset; krnCol < offset + 1; krnCol++){
                    //Max value under the kernel
                    if(kernel[offset + krnRow][offset + krnCol]*
                            imgIn[idxRow + krnRow][idxCol + krnCol] > max)        {
                        max = imgIn[idxRow + krnRow][idxCol + krnCol];
                    }
                }
            }

            imgOut[idxRow][idxCol] = max;
        }
    }

    FILE *fp = fopen("ps_dil.log", "w");

    for(int idxRow = 0; idxRow < IMG_ROWS; idxRow++){
        for(int idxCol = 0; idxCol < IMG_COLS; idxCol++){
            fprintf(fp, "%d ", imgOut[idxRow][idxCol]);
        }
    }

    fclose(fp);

    return 0;

}

如您所见,我的输入图像位于imgIn数组中,而结构化元素位于kernel数组中,其中所有元素均设置为1。而且,我没有处理边框,使其像素值为0。

输入图像来自一维数组,该数组是通过以下matlab脚本生成的:

function saveAsCArray( I )
[nRows, nCols] = size(I);

cFile = fopen('image.h', 'w');
fprintf(cFile, 'unsigned char img[%d] = {', (nRows*nCols));

for row = 1:nRows
    for col = 1:nCols
        if row == nRows && col == nCols
            fprintf(cFile, '%d};', I(row, col));
        else
            fprintf(cFile, '%d,', I(row, col));
        end
    end
    fprintf(cFile, '\n');
end

fclose(cFile);

结束

使用以下脚本将输出日志文件转换回图像:

function intFileToImg( fileName, imgName, imgSizeRow, imgSizeCol)

    A = dlmread(fileName);
    A = uint8(A);
    A = reshape(A, imgSizeRow, imgSizeCol);
    A = rot90(A,3);
    I = mat2gray(A);
    I = flip(I, 2);
    imwrite(I, imgName);
    imshow(I);

end
我以输出像素接收内核窗口内最大像素值的方式实现了扩展。问题是我的输出图像显示了一些奇怪的重复,我真的认为这与某些索引错误有关。不幸的是,我无法弄清楚自己在做什么。

这是我的输入内容:

enter image description here

这是我使用5x5内核的输出: enter image description here

这是我使用3x3内核的输出: enter image description here

请,有人可以看一下代码并帮助我找出我做错了什么吗?

1 个答案:

答案 0 :(得分:1)

将图像转换为线性阵列时,可以连续写入每一行(行长)。但是,当将线性数组转换回图像时,请使用以下顺序:

A = reshape(A, imgSizeRow, imgSizeCol);
A = rot90(A,3);
I = mat2gray(A);
I = flip(I, 2);

MATLAB主要用于列,这就是为什么您需要在其中使用rot90的原因。但是正因为如此,您还应该切换行和列的大小:

A = reshape(A, imgSizeCol, imgSizeRow).';
I = mat2gray(A);

还请注意,rot90 + flip与转置矩阵相同,这是您实际需要的(从行优先矩阵到列优先矩阵,您需要交换这两个维度,这就是移调的作用。


您还应该在C程序中修复以下两行代码:

#define IMG_ROWS 1280
#define IMG_COLS 1024

您发布为输入的图像具有502行和622列。

我创建了这个MATLAB脚本:

% Load image
I = imread('https://i.stack.imgur.com/0u3bK.png');
I = I(:,:,1); % Keep only the first channel

% Write image
[nRows, nCols] = size(I);
cFile = fopen('image.h', 'w');
fprintf(cFile, '#define IMG_ROWS %d\n', nRows);
fprintf(cFile, '#define IMG_COLS %d\n', nCols);
fprintf(cFile, 'unsigned char img[%d] = {', (nRows*nCols));
fprintf(cFile, '%d,', I.'); % Note transpose!
fprintf(cFile, '};\n');
fclose(cFile);

% Compile & run C code
!gcc so.c -o so
!./so

% Load output
A = dlmread('ps_dil.log');
A = uint8(A);
A = reshape(A, nCols, nRows).';
imshow(A);

文件so.c是您发布的C代码,但是删除了定义IMG_ROWSIMG_COLS的两行。我在这里创建的image.h文件将这两行写入。这是输出,完美的扩张:

output of code above