Adversarial Learning for Semi-supervised Semantic Segmentation, BMVC 2018
Alternatives To Advsemiseg
Project NameStarsDownloadsRepos Using ThisPackages Using ThisMost Recent CommitTotal ReleasesLatest ReleaseOpen IssuesLicenseLanguage
Labelme9,8968811 days ago177March 03, 202267otherPython
Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).
Awesome Semantic Segmentation8,065
2 years ago13
:metal: awesome-semantic-segmentation
Segmentation_models.pytorch6,98123414 hours ago10November 18, 202126mitPython
Segmentation models with pretrained backbones. PyTorch.
Paddleseg6,652216 hours ago8April 20, 2022184apache-2.0Python
Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image Matting, 3D Segmentation, etc.
Pytorch Unet6,465
19 days ago49gpl-3.0Python
PyTorch implementation of the U-Net for image semantic segmentation with high quality images
Mmsegmentation5,458220 hours ago30July 01, 2022292apache-2.0Python
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
Gluon Cv5,42215442 months ago1,514July 07, 202261apache-2.0Python
Gluon CV Toolkit
Hrnet Semantic Segmentation2,714
4 months ago151otherPython
The OCR approach is rephrased as Segmentation Transformer: This is an official implementation of semantic segmentation for HRNet.
Awesome Semantic Segmentation Pytorch2,399
3 months ago114apache-2.0Python
Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
Imgclsmob2,3994a year ago67September 21, 20216mitPython
Sandbox for training deep learning networks
Alternatives To Advsemiseg
Select To Compare

Alternative Project Comparisons

Adversarial Learning for Semi-supervised Semantic Segmentation

This repo is the pytorch implementation of the following paper:

Adversarial Learning for Semi-supervised Semantic Segmentation
Wei-Chih Hung, Yi-Hsuan Tsai, Yan-Ting Liou, Yen-Yu Lin, and Ming-Hsuan Yang
Proceedings of the British Machine Vision Conference (BMVC), 2018.

Contact: Wei-Chih Hung (whung8 at ucmerced dot edu)

The code are heavily borrowed from a pytorch DeepLab implementation (Link). The baseline model is DeepLabv2-Resnet101 without multiscale training and CRF post processing, which yields meanIOU 73.6% on the VOC2012 validation set.

Please cite our paper if you find it useful for your research.

  author = {W.-C. Hung and Y.-H. Tsai and Y.-T. Liou and Y.-Y. Lin and M.-H. Yang},
  booktitle = {Proceedings of the British Machine Vision Conference (BMVC)},
  title = {Adversarial Learning for Semi-supervised Semantic Segmentation},
  year = {2018}


  • pytorch >= 0.2 (We only support 0.4 for evaluation. Will migrate the code to 0.4 soon.)
  • python-opencv >=3.4.0 (3.3 will cause extra GPU memory on multithread data loader)


  • Clone this repo
git clone
  • Place VOC2012 dataset in AdvSemiSeg/dataset/VOC2012. For training, you will need the augmented labels (Download). The folder structure should be like:

Testing on VOC2012 validation set with pretrained models

python --pretrained-model semi0.125 --save-dir results

It will download the pretrained model with 1/8 training data and evaluate on the VOC2012 val set. The colorized images will be saved in results/ and the detailed class IOU will be saved in results/result.txt. The mean IOU should be around 68.8%.

  • Available --pretrained-model options: semi0.125, semi0.25, semi0.5 , advFull.

Example visualization results

Training on VOC2012

python --snapshot-dir snapshots \
                --partial-data 0.125 \
                --num-steps 20000 \
                --lambda-adv-pred 0.01 \
                --lambda-semi 0.1 --semi-start 5000 --mask-T 0.2

The parameters correspond to those in Table 5 of the paper.

To evaluate trained model, execute the following:

python --restore-from snapshots/VOC_20000.pth \
                       --save-dir results


  • 07/24/2018: Update BMVC results
Popular Segmentation Projects
Popular Semantic Segmentation Projects
Popular Machine Learning Categories
Related Searches

Get A Weekly Email With Trending Projects For These Categories
No Spam. Unsubscribe easily at any time.
Deep Learning
Computer Vision
Semantic Segmentation
Semi Supervised Learning
Adversarial Learning