Segmentation Zoo
This library contains the code to perform segmentation with the following networks:
FCN-8
FCN-16
FCN-32
U-Net
SegNet
The data should be stored as follows:
Within Segmentation_parameters.py
, specify the data_root
. Within this directory store Images, Labels and Weights as the following directories:
Images
Labels
Weights
Labels are stored as RGB images with 3 channels, where the first channel gives the positive class and the second channel gives the negative class. Weights have one channel. Corresponding Image, Label and Weight files must have the same name. There must be the same amount of data within each folder and all data should have the same x,y dimensions.
To start training type within the terminal:
python Segmentation_main.py train