Starred repositories
liver segmentation using deep learning
This repository contains the implementation of the paper -- KNOT: Knowledge Distillation using Optimal Transport for Solving NLP Tasks
Channel-wise Distillation for Semantic Segmentation
[CVPR-2022] Official implementations of CIRKD: Cross-Image Relational Knowledge Distillation for Semantic Segmentation and implementations on Cityscapes, ADE20K, COCO-Stuff., Pascal VOC and CamVid.
Collection of common code that's shared among different research projects in FAIR computer vision team.
Count the MACs / FLOPs of your PyTorch model.
This is the implementation for the ICME-2023 paper (Adaptive Multi-Teacher Knowledge Distillation with Meta-Learning).
AMTML-KD: Adaptive Multi-teacher Multi-level Knowledge Distillation
This is the implementation for the ICASSP-2022 paper (Confidence-Aware Multi-Teacher Knowledge Distillation).
Multi-student knowledge distillation, which employs a multiple student model to bridge the gap between the data distribution and the student meta architecture.
Improving Convolutional Networks via Attention Transfer (ICLR 2017)
Intra-class Feature Variation Distillation for Semantic Segmentation (ECCV 2020)
Official PyTorch implementation of "Lightweight Deep CNN for Natural Image Matting via Similarity-Preserving Knowledge Distillation" (IEEE Signal Processing Letters 2020)
This is an official implementation of KDAS for Knowledge Distillation Polyp Segmentation (ICME 2024)
[TMI'20] Unpaired Multi-modal Segmentation via Knowledge Distillation
Advancing Medical Image Segmentation Through Multi-Task and Multi-Scale Contrastive Knowledge Distillation
Knowledge Distillation: CVPR2020 Oral, Revisiting Knowledge Distillation via Label Smoothing Regularization
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
Implementation of SegFormer in PyTorch
[TMI'22] "AADG: Automatic Augmentation for Domain Generalization on Retinal Image Segmentation".
Adaptive Affinity Fields for Semantic Segmentation
Pytorch implementation of various Knowledge Distillation (KD) methods.
PyTorch implementation of over 30 realtime semantic segmentations models, e.g. BiSeNetv1, BiSeNetv2, CGNet, ContextNet, DABNet, DDRNet, EDANet, ENet, ERFNet, ESPNet, ESPNetv2, FastSCNN, ICNet, LEDN…
[MICCAI'22] Joint Class-Affinity Loss Correction for Robust Medical Image Segmentation with Noisy Labels
TABS: Transformer-based Automated Brain Segmentation
[ICASSP 2023] Prototype Knowledge Distillation for Medical Segmentation with Missing Modality