Here are a few pointers which may be of help:
- The function “ndims()” ignores all the trailing singleton dimensions of the input matrix. Hence it ignores the third singleton dimension and returns 2 when the input is a 2-D matrix.
- One possible workaround is to use the “permute()” function as follows:
if ndims(permute(I, [3 1 2])) ~= 3 || Isize ~= networkChannelSize
error(message('vision:rcnn:invalidInputImageChannelSize', Isize, networkChannelSize));
end
If “I" is of size m x n, the permute operation changes the size of the matrix to 1 x m x n and “ndims()” returns 3.
This method may create an error if “I” is a 4-D or higher dimensional matrix.
- Else you may include another “if” condition which checks whether “ndims(I)” is equal to 2 or not.
Hope this helps!!
