ets-research-repositories Goto Github PK

3d-skipdenseseg

aet

Auto-Encoding Transformations (AETv1), CVPR 2019

ccnn

[ICCV 2015] Framework for optimizing CNNs with linear constraints for Semantic Segmentation

clustergan

Pytorch Implementation of ClusterGAN (arXiv:1809.03627)

config-parser-battery

crst

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

dccs

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

deepgrabcut-pytorch

Deep GrabCut in PyTorch

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

learn_noisy_labels_medical_images

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

monai

AI Toolkit for Healthcare Imaging

nilearn

Machine learning for NeuroImaging in Python

openselfsup

Self-Supervised Learning Toolbox and Benchmark

organsegrstn_pytorch

PyTorch implementation of OrganSegRSTN - CVPR 2018

pcl

PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"

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)

surface-distance

Library to compute surface distance based performance metrics for segmentation tasks.