Semantic_segmentation

Data

• Link to download dataset used for training (https://drive.google.com/drive/folders/1NChdMcWCy_pGADKQLb8A3_YW1NjtZM0c?usp=share_link)
• Link to download dataset used for test ( https://drive.google.com/drive/folders/1kH96RBCpEI6kP-NcoQpPBBBa4urTlepK?usp=share_link)
• Link to download best checkpoint (https://drive.google.com/drive/folders/1XjGzut4a3JIhq-Nkbg4-choc1ImyaWpF?usp=share_link)

Requirements

• pytorch
• torchvision

Training

• python train.py

Prediction

• python predict.py -i input image -o output image

Reference: (https://github.com/zhixuhao/unet.git)