ETS-Research-Repositories

DCCS

Official PyTorch implementation for ECCV'20 paper: Deep Image Clustering with Category-Style Representation

Learn_Noisy_Labels_Medical_Images

[NeurIPS 2020] Disentangling Human Error from the Ground Truth in Segmentation of Medical Images

PHiSeg-code

Tensorflow Code for "PHiSeg: Capturing Uncertainty in Medical Image Segmentation", Proc. MICCAI 2019

PICA

Official Pytorch Implementation for CVPR'20 paper: Deep Semantic Clustering by Partition Confidence Maximisation

PlotNeuralNet

Latex code for making neural networks diagrams

PMMs

Prototype Mixture Models for Few-shot Semantic Segmentation

Probabilistic-Unet-Pytorch

A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch

PyContrast

PyTorch implementation of Contrastive Learning methods; List of awesome-contrastive-learning papers

pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

semseg

Semantic Segmentation in Pytorch

smile-mi-estimator

PyTorch implementation for the ICLR 2020 paper "Understanding the Limitations of Variational Mutual Information Estimators"

submitit

Python 3.6+ toolbox for submitting jobs to Slurm

SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

ummkd

[TMI'20] Unpaired Multi-modal Segmentation via Knowledge Distillation

3D-SkipDenseSeg

AET

Auto-Encoding Transformations (AETv1), CVPR 2019

clusterGAN

Pytorch Implementation of ClusterGAN (arXiv:1809.03627)

Config-Parser-battery

CRST

Code for <Confidence Regularized Self-Training> in ICCV19 (Oral)

demi_mi_estimator

PyTorch demo of "DEMI: Discriminative Estimator of Mutual Information"

detectron2

Detectron2 is FAIR's next-generation research platform for object detection and segmentation.

EfficientFormer

ESPNetv2

A light-weight, power efficient, and general purpose convolutional neural network

MONAI

AI Toolkit for Healthcare Imaging

nilearn

Machine learning for NeuroImaging in Python

OpenSelfSup

Self-Supervised Learning Toolbox and Benchmark

PCL

PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"

torchio

Tools for loading, augmenting and writing 3D medical images on PyTorch.

U-HVED

Hetero-Modal Variational Encoder-Decoder for Joint Modality Completion and Segmentation

ViT-pytorch

Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)