Walterkd

Inf-DiT

Official implementation of Inf-DiT: Upsampling Any-Resolution Image with Memory-Efficient Diffusion Transformer

trader

交易模块

ALDM

Adaptive Latent Diffusion Model for 3D Medical Image

MambaOut

MambaOut: Do We Really Need Mamba for Vision?

DiT-3D

🔥🔥🔥Official Codebase of "DiT-3D: Exploring Plain Diffusion Transformers for 3D Shape Generation"

bitsandbytes

Accessible large language models via k-bit quantization for PyTorch.

llm.c

LLM training in simple, raw C/CUDA

dpm-solver

Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

med-ddpm

Mamba-UNet

Mamba-UNet Zoo

DisC-Diff

【MICCAI 2023, Early accept】DisC-Diff: Disentangled Conditional Diffusion Model for Multi-Contrast MRI Super-Resolution

3D-VQ-VAE-2

3D VQ-VAE-2 for high-resolution CT scan synthesis

ResShift

ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shifting (NeurIPS 2023 Spotlight)

medicaldiffusion

Medical Diffusion: This repository contains the code to our paper Medical Diffusion: Denoising Diffusion Probabilistic Models for 3D Medical Image Synthesis

PerceptualSimilarity

LPIPS metric. pip install lpips

mri_super_resolution

MRI Super Resolution with a Deep 3D Resnet Variant using Perceptual Loss

BasicSR

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.

Real-ESRGAN

Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.

gdemo_pancreassegmentation

demo of a vae model for pancreas segmentation.

DWIinpainting

Reconstructing cropped data in 3D diffusion MRI images using variational autoencoders

Variational-AutoEncoder

An Pytorch Implementation of Variational AutoEncoder for 3D MRI Brain Image

Palette-Image-to-Image-Diffusion-Models

Unofficial implementation of Palette: Image-to-Image Diffusion Models by Pytorch

Image-Super-Resolution-via-Iterative-Refinement

Unofficial implementation of Image Super-Resolution via Iterative Refinement by Pytorch

VascuSynth

Vascular Tree Synthesis Software. You will need CMake and ITK to run VascuSynth. Read the paper in the Insight Journal for instructions.

ArtiFusion

MICCAI2023: Artifact Restoration in Histology Images with Diffusion Probabilistic Models

MedSegDiff

Medical Image Segmentation with Diffusion Model

SADIR-Shape-Aware-Difffusion-Models-for-3D-Image-Reconstruction

DiffPIR

"Denoising Diffusion Models for Plug-and-Play Image Restoration", Yuanzhi Zhu, Kai Zhang, Jingyun Liang, Jiezhang Cao, Bihan Wen, Radu Timofte, Luc Van Gool.

DPIR

Plug-and-Play Image Restoration with Deep Denoiser Prior (IEEE TPAMI 2021) (PyTorch)