Skip to content

Convolutional Neural Networks for Image Classification

In this Computer Vision coursework, I explored the application of convolutional neural networks for image classification tasks, specifically in the domain of fish classification for precision fishing. I developed an image classification model using PyTorch, that can classify fish species given input images. This coursework contains two tasks: 1. Train a model to classify various fish species. 2. Finetuning the last layer of the trained model to adapt to some new species. The trained model and the finetuned model both reach the accuracy of 99%, on the 800 test images.