Car Image Segmentation project

This project implements the U-Net architecture to train a model that can segment car images. The dataset used for the model is from the Carvana Image Segmentation Challenge Kaggle Page, which can be found here. I worked with a reduced dataset that cut down on alot of images so the model could be trained on a reasonable time. You can find the version of that dataset on my github linked below. The U-Net architecture can be referenced below:

Final Segmentation Result

Dice Score: 0.986

Code for the Model including datasets