Stars
This is a summary of research on general-purpose Medical Image Restoration. Please raise an issue if you suggest new qualified project.
PyTorch implementation of JiT https://arxiv.org/abs/2511.13720
[AAAI 2026] EndoIR: Degradation-Agnostic All-in-One Endoscopic Image Restoration via Noise-Aware Routing Diffusion
Universal Image Restoration Pre-training via Masked Degradation Classification
PyTorch implementation of Directional Adaptive Shuffle Mamba (MICCAI 2025) for MRI SR, CT denoising & PET synthesis
[NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.
[ICLR2025 Spotlight🔥] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters
SurgClean Benchmark for Surgical Image Restoration.
[MICCAI 2025] FEAT：Full-Dimensional Efficient Attention Transformer for Medical Video Generation.
The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)
Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"
This is a pytorch implementation of Denoising Diffusion Implicit Models
Awesome radiology report generation and image captioning papers.
[ICCV 2025] FoundIR: Unleashing Million-scale Training Data to Advance Foundation Models for Image Restoration
ECCV2024：A Comparative Study of Image Restoration Networks for General Backbone Network Design
PromptIR: Prompting for All-in-One Blind Image Restoration [NeurIPS 2023]
Official PyTorch implementation of "EdgeSAM: Prompt-In-the-Loop Distillation for On-Device Deployment of SAM"
CVPR 2024-Video Super-Resolution Transformer with Masked Inter&Intra-Frame Attention
[NeurIPS2024] Causal Context Adjustment Loss for Learned Image Compression
[ICLR 2025] An Intelligent Agentic System for Complex Image Restoration Problems
[IEEE TPAMI] A Survey on All-in-One Image Restoration: Taxonomy, Evaluation and Future Trends
An up-to-date list of works on Multi-Task Learning
(CVPR2024) Official implementation of paper: "Selective Hourglass Mapping for Universal Image Restoration Based on Diffusion Model"
This repository contains the code used to train an LSTM convolutional neural network to perform fly heart segmentation
ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shifting (NeurIPS@2023 Spotlight, TPAMI@2024)
Helical to fan-beam CT geometry rebinning and differentiable FBP of DICOM-CT-PD projections
ACMMM24 - Learning A Low-Level Vision Generalist via Visual Task Prompt Arxiv - Exploring Scalable Unified Modeling for General Low-Level Vision