PyTorch Vision Toolbox not only for deep-clustering

clustering, framework, pytorch, segmentation, semi-supervised-learning, toolbox
pip install deepclustering==0.0.3



PyTorch Vision toolbox not only for deep-clustering


I still use this repo for research propose. I update some modules frequently to make the framework flexible enough.

This repo contains the base code for a deep learning framework using PyTorch, to benchmark algorithms for various dataset. The current version supports MNIST, CIFAR10, SVHN and STL-10 for semisupervised and unsupervised learning. ACDC, Promise12, WMH and so on are supported as segmentation counterpart.


  • Powerful cmd parser using yaml module, providing flexible input formats without predefined argparser.
  • Automatic checkpoint management adapting to various settings
  • Automatic meter recording and experimental status plotting using matplotlib and threads
  • Various build-in loss functions and help tricks and assert statements frequently used in PyTorch Framework, such as disable_tracking_bn, ema, vat, etc.
  • Various post-processing tools such as Viewer for Medical image segmentations, multislice_viwers for 3D dataset real-time debug and report script for experimental summaries.
  • Extendable modules for rapid development.

Several projects are benefited from this scalable framework, builing top on this, including:

  • DeepClustering implemented for
  • SemiSupervised classification for
  • Semi-Supervised Learning by Augmented Distribution Alignment,
  • Virtual Adversarial Training: A Regularization Method for Supervised and Semi-Supervised Learning,
  • Temporal Ensembling for Semi-Supervised Learning,
  • Mean teachers are better role models: Weight-averaged consistency targets improve semi-supervised deep learning results
  • SemiSupervised Segmentation for
  • Discretely-constrained CNN for

They are examples how to develop research framework with the assistance of our proposed deep-clustering-toolbox.


Several papers have been implemented based on this framework. I store them in the playground folder. The papers include:


git clone
cd deep-clustering-toolbox  
python setup install # for those who do not want to make changes immediately.
# or
python setup develop # for those who want to modify the code and make the impact immediate.

Or very simply

pip install deepclustering


If you feel useful for your project, please consider citing this work.

  title={Deep Co-Training for Semi-Supervised Image Segmentation},
  author={Peng, Jizong and Estradab, Guillermo and Pedersoli, Marco and Desrosiers, Christian},
  journal={arXiv preprint arXiv:1903.11233},