In this problem the challenge is to build the Matrix Form equivalent of the function `conv2()` of MATLAB.
The function to be built will generate a sparse matrix `mK` which is the convolution matrix of the 2D Kernel 'mH' (Which is an input to the function). The input to the function is the 2D Kernel and dimensions of the image to apply the convolution upon and the shape of the convolution ('full', 'same', 'valid' as in 'conv2()').
The output sparse matrix should match the output of using 'conv2()' on the same image using the same convolution shape.
For instance:
CONVOLUTION_SHAPE_FULL = 1; CONVOLUTION_SHAPE_SAME = 2; CONVOLUTION_SHAPE_VALID = 3;
numRowsImage = 100; numColsImage = 80;
numRowsKernel = 7; numColsKernel = 5;
mI = rand(numRowsImage, numColsImage); mH = rand(numRowsKernel, numColsKernel);
maxThr = 1e-9;
%% Full Convolution
convShape = CONVOLUTION_SHAPE_FULL;
numRowsOut = numRowsImage + numRowsKernel - 1; numColsOut = numColsImage + numColsKernel - 1;
mORef = conv2(mI, mH, 'full'); mK = CreateImageConvMtx(mH, numRowsImage, numColsImage, convShape); mO = reshape(mK * mI(:), numRowsOut, numColsOut);
mE = mO - mORef; assert(max(abs(mE(:))) < maxThr);
The test case will examine all 3 modes. Try to solve it once with very clear code (No vectorization tricks) and then optimize.
A good way to build the output sparse function is using:
mK = sparse(vRows, vCols, vVals, numElementsOutputImage, numElementsInputImage);
Look for the documentation of `sparse()` function for more details.
22973 Solvers
Is my wife right? Now with even more wrong husband
1145 Solvers
54 Solvers
28 Solvers
Cody Computer Part 4 : Are you a morning Cody player Addicted ?
66 Solvers
Problem Tags