Under Construction!
Final Project for my machine learning class. I implemented the AGAR dataset from the Automated Bacterial Colony Segmentation project and utilized transfer learning in addition to training my own model using PyTorch to classify the 5 different bacterial colonies in the dataset. I suceded in gaining satisfactory results from transfer learning and excellent results from training my own model.
My main takeaways from this project were learning about how to segment a dataset for the creation of robust models and learning the basics of creating a classifier using a convolutional neural network.