tanphan07 / UNet_CityScapes

UNet Model for City-Scapes Dataset

Requirements

Install all requirements in requirements.txt file:

pip3 install -r requirements.txt

Train model

!! Before runing this command, you must change mode of this bash file by :

chmod +x run_train.sh

Training model by running this command

./run_train.sh

Evaluation

!! Before runing this command, you must change mode of this bash file by :

chmod +x run_test.sh

Evaluate model by running this command

./run_test.sh

Inference

!! Before runing this command, you must change mode of this bash file by :

chmod +x run_infer.sh

./run_infer.sh

Dataset Architecture

CityScapes
|----gtFine
|        |-- test
|        |-- train
|        |-- val
|    
|---- leftImg8bit
         |-- test
         |-- train
         |-- val

Checkpoint Model

Checkpoint model is saved in saved/models/UNet/DiceLoss_475_v2/checkpoint-epoch480.pth You can download checkpoint from here Google Drive and put it to the repository.