/U-net_Cityscapes

Deep Learning for Computer Vision 2022 Summer Program - Segmentation and Uncertainty

Primary LanguageJupyter NotebookMIT LicenseMIT

U-net Cityscapes using tensorflow

Deep Learning for Computer Vision 2022 Summer Program - Segmentation and Uncertainty

This repository stores everything of the team project in Deep Learning for Computer Vision 2022 Summer Program (麻省理工学院前沿学科项目:深度学习应用于计算机视觉). It contains the code implementation of U-net at Cityscapes Dataset using tensorflow framework.

You are free to clone this repo and modify everything:wink:. Just type in the following code in your terminal:

git clone https://github.com/JohnG-mit/U-net_Cityscapes.git

If you want to train the network by yourself, I provide the dataset and model checkpoint in Release page. The training environment is clearly described in the jupyter notebook.