Backbone of Computer Vision

CNNs are a type of neural network made for images. They work by scanning small parts of an image, not the whole thing at once.

In this page I will be going through the basic intuition of CNNs, the forward pass, and how convolution + pooling work together.

— Alexi


Convolution → RELU → Pooling

Source: https://medium.com/hackernoon/visualizing-parts-of-convolutional-neural-networks-using-keras-and-cats-5cc01b214e59

Types of Convolution Layer:

1D Convolution Layer:

image.png

2D Convolution Layer:

image.png

Code and Explaination for Conv2D:

nn.Conv2d(
				in_channels=1, 
				out_channels=32, 
				kernel_size=3, 
				stride=1, 
				padding=1
				)

image.png