edualgo / Grokking-Machine-Learning

This repo aims to contain different machine learning use cases along with the descriptions to the model architectures
https://github.com/Abhijit2505/Grokking-Machine-Learning/wiki
MIT License
63 stars 49 forks source link

Image Segmentation using UNet #121

Closed infiniteoverflow closed 3 years ago

infiniteoverflow commented 3 years ago

Dataset : https://www.kaggle.com/tapakah68/segmentation-full-body-mads-dataset

We will be using the UNet Architecture to implement a model that can extract an image from its background.

Abhijit2505 commented 3 years ago

@infiniteoverflow please get started.