Convolution Neural Network Image Classifier


Following the initial image classifier built using the feed-forward architecture the convolution neural networks are the next step on the quest of surpassing human-level accuracy. In essence, the nodes making up the layers of CNNs are not fully connected with nodes adjacent to them, but are rather divided into distinct sets covered by the filter, or the convolution kernel (hence the name).


Such an approach allows the network to “compress” tensors and thus extract and (hopefully) learn important details about the images it is trying to classify. In result the network is both more accurate and also faster to train from the get-go. Also, it’s important to mention that the CNN in this case is actually AlexNet, historically important first network that attempted to classify images as described.

In case you’re interested in an entry level PyTorch deep learning textbook I wholeheartedly recommend the following book by Ian Pointer:

Jupyter notebook of the model may be found on GitHub:

Also, the same notebook is available on Google Colab (where it can be tested it against the GPU available there):

Leave a Reply