Class for doing convolution over images.
Class for doing convolution over images.Suppose you put n_input_planes images in each input frame. The images are in one big vector: each input frame has a size of n_input_planes*input_height*input_width. (image after image). Thus, n_inputs = n_input_planes*input_height*input_width.
Then, for each output planes, it computes the convolution of all input image planes with a kernel of size k_w*k_w*n_input_planes.
The output image size is computed in the constructor and put in output_height and output_width. n_outputs = n_output_planes*output_height*output_width.
Note that, depending of the size of your kernel, several (last) columns or rows of the input image could be lost.
Note also that no non-linearity is applied in this layer.
Alphabetic index HTML hierarchy of classes or Java