#
Starred repositories
PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
[T-PAMI 2025] SceneTracker: Long-term Scene Flow Estimation Network
[CVPR 2024 Highlight] Official PyTorch implementation of SpatialTracker: Tracking Any 2D Pixels in 3D Space
[CVPR 2024 highlight] Progressive Divide-and-Conquer via Subsampling Decomposition for Accelerated MRI, PyTorch Implementation
Joint Atlas Building and Diffeomorphic Registration Learning with Pairwise Alignment
Code for CVPR 2022 paper GANSeg: Learning to Segment by Unsupervised Hierarchical Image Generation
[IEEE TMI'24] Self-Supervised Cyclic Diffeomorphic Mapping for Soft Tissue Deformation Recovery in Robotic Surgery Scenes
[CVPR 2024] VoCo: A Simple-yet-Effective Volume Contrastive Learning Framework for 3D Medical Image Analysis
Semi Supervised Learning for Medical Image Segmentation, a collection of literature reviews and code implementations.
PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
Official implementation for "QS-Attn: Query-Selected Attention for Contrastive Learning in I2I Translation" (CVPR 2022)
[CVPR 2022] C2AM: Contrastive learning of Class-agnostic Activation Map for Weakly Supervised Object Localization and Semantic Segmentation
Count the MACs / FLOPs of your PyTorch model.
tracking medical datasets, with a focus on medical imaging
[NeurIPS 2023] Release LMV-Med pre-trained models
Code for TKDE paper "Self-supervised learning on graphs: Contrastive, generative, or predictive"
A fast medical imaging analysis library in Python with algorithms for registration, segmentation, and more.
[CVPR 2024] Intraoperative 2D/3D registration via differentiable X-ray rendering
Code for the paper "Preserving Volume for Unsupervised Registration" (ICCV 2023 Poster)
MambaMorph: a Mamba-based Framework for Medical MR-CT Deformable Registration
"Structure-Aware Sparse-View X-ray 3D Reconstruction" (CVPR 2024) - A Toolbox for CT reconstruction and X-ray Novel View Synthesis
[CVPR 2024] Generalizable Tumor Synthesis - Realistic Synthetic Tumors in Liver, Pancreas, and Kidney
Diffusion Models in Medical Imaging (Published in Medical Image Analysis Journal)