Implementation of several state of the art segmentation method for binary tasks in PyTorch

Here are the implemented models:

• [FCN8]
• [FCN16]
• [FCN32]
• [SegNet]
• [PSPNet]
• [UNet]
• [Residual UNet]
• [DUC]
• [duchdc]
• [LinkNet]
• [FusionNet]
• [GCN]

Usage

For quick hints about commands:

python main.py -h

Modify the condiguration in the settings.py file

Training

After customizing the settings.py, use the following command to start training

python main.py --cuda train

Evaluation

For evaluation, put all your test images in a folder and set path in the settings.py. Then run the following command:

python main.py --cuda eval

The results will be place in the results directory

This doc will be improved