TensorFlow-based semantic segmentation codes.
- Fully Convolutional Network model.
- Dilated Network model.
- Pyramid Scene Parsing Network model.
- Cityscapes dataset.
- VOC2012 dataset.
This repo only assumes .tfrecord format data.
The .tfrecord data should be created by scripts/write_tfrecord_${data_name}.py.
You can train model by scripts/trainer_${model_name}_${data_name}.py.
Example parameters are defined in scripts/params/train_params_${model_name}_${data_name}.yaml.
You can run model inference by scripts/predictor_${model_name}_${data_name}.py.
Example parameters are defined in scripts/params/predict_params_${model_name}_${data_name}.yaml.