Three Stage Self-training Semantic Segmentation

This repository is about our work "A Three-Stage Self-Training Framework for Semi-Supervised Semantic Segmentation"

This repository is tested under

python 3.7.3
pytorch 1.7.1
numpy 1.19.2
pillow 6.0.0
scikit-image 0.15.0
torchvision 0.8.2

Cityscape
Pascal voc 2012

Cityscapes:

Stage1: python main.py 1
Stage2: python main.py 2
Stage3: python main.py 3

Pascal voc:

Stage1: python main.py 4
Stage2: python main.py 5
Stage3: python main.py 6

R. Ke, A. Aviles-Rivero, S. Pandey, S. Reddy, and C. B. Schönlieb. A Three-Stage Self-Training Framework for Semi-Supervised Semantic Segmentation. IEEE Transactions on Image Processing. 31 (2022): 1805-1815

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
architectures		architectures
data		data
datapipe		datapipe
tools		tools
README.md		README.md
apply_rand.py		apply_rand.py
cf2.py		cf2.py
evaluation.py		evaluation.py
main.py		main.py
para1.py		para1.py
randaugment_seg.py		randaugment_seg.py
script_lib.py		script_lib.py
semantic_segmentation.cfg		semantic_segmentation.cfg
settings.py		settings.py