weihuang-cs

kornia

Geometric Computer Vision Library for Spatial AI

MONAI

AI Toolkit for Healthcare Imaging

detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

Semi-supervised-learning

A Unified Semi-Supervised Learning Codebase (NeurIPS'22)

DCNv4

[CVPR 2024] Deformable Convolution v4

Mamba-UNet

Mamba-UNet Zoo

STU-Net

The largest pre-trained medical image segmentation model (1.4B parameters) based on the largest public dataset (>100k annotations), up until April 2023.

volumentations

Library for 3D augmentations

core-pytorch-utils

Yet another PyTorch Trainer and some core components for deep learning.

Awesome-Implicit-Neural-Representations-in-Medical-imaging

[ICCV 2023] A curated list of resources on implicit neural representations in Medical Imaging

ICT

Code for reproducing ICT (published in Neural Networks 2022, and in IJCAI 2019)

Fast-Symmetric-Diffeomorphic-Image-Registration-with-Convolutional-Neural-Networks

Fast Symmetric Diffeomorphic Image Registration with Convolutional Neural Networks

deedsBCV

new and more efficient version for 3D discrete deformable registration, reaching the highest accuracy in several benchmarks

mmFormer

[MICCAI 2022] The official code for "mmFormer: Multimodal Medical Transformer for Incomplete Multimodal Learning of Brain Tumor Segmentation"

CoCoNet

[IJCV 2023] CoCoNet: Coupled Contrastive Learning Network with Multi-level Feature Ensemble for Multi-modality Image Fusion

NeRP

Implicit Neural Representation Learning With Prior Embedding for Sparsely Sampled Image Reconstruction

kmax-deeplab

a PyTorch re-implementation of ECCV 2022 paper based on Detectron2: k-means mask Transformer.

NeSVoR

NeSVoR is a package for GPU-accelerated slice-to-volume reconstruction.

ArSSR

[IEEE J-BHI] An Arbitrary Scale Super-Resolution Approach for 3D MR Images using Implicit Neural Representation

ICNet

Inverse-Consistent Deep Networks for Unsupervised Deformable Image Registration

ProCA

[ECCV 2022] Prototypical Contrast Adaptation for Domain Adaptive Semantic Segmentation

mmengine-template

NexToU

NexToU: Efficient Topology-Aware U-Net for Medical Image Segmentation

long-tailed-ood-detection

Official implementation for "Partial and Asymmetric Contrastive Learning for Out-of-Distribution Detection in Long-Tailed Recognition" (ICML'22 Long Presentation)

SVoRT

SVoRT: Iterative Transformer for Slice-to-Volume Registration in Fetal Brain MRI

DoCR

Code for [MICCAI 2022] Domain Specific Convolution and High Frequency Reconstruction based Unsupervised Domain Adaptation for Medical Image Segmentation.

SAINT

SAINT: Spatially Aware Interpolation NeTwork for Medical Slice Synthesis (CVPR2020)

SAA

Code and Model For SAA

2023_NeurIPS_FlatMatch

MADGNet

CVPR2024 Accepted Paper