Skip to content

Latest commit

 

History

History
19 lines (14 loc) · 516 Bytes

README.md

File metadata and controls

19 lines (14 loc) · 516 Bytes

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