Skip to content

sudarshan-kamath/U-Net-implementation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Implementation of the U-Net segmentation network using PyTorch

The main goal is to create a U-Net network from scratch by understanding the paper.

Further goals:

  • Modification of the MobileNetV2 SSD for segmentation in realtime.
  • Training of the UNet on Medical imaging datasets from Kaggle and the Traffic datasets.
  • Other optimisations if possible

Paper link from https://arxiv.org/abs/1505.04597

Tasks

  • Code the network
  • Dataset
  • Training
  • Results
  • Video
  • Explanation

About

No description or website provided.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages