我正在尝试使用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
我以输出像素接收内核窗口内最大像素值的方式实现了扩展。问题是我的输出图像显示了一些奇怪的重复,我真的认为这与某些索引错误有关。不幸的是,我无法弄清楚自己在做什么。
这是我的输入内容:
请,有人可以看一下代码并帮助我找出我做错了什么吗?
答案 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_ROWS
和IMG_COLS
的两行。我在这里创建的image.h
文件将这两行写入。这是输出,完美的扩张: