← all repositories

Janspiry/Palette-Image-to-Image-Diffusion-Models

PyTorch implementation of Palette, a diffusion model for image-to-image tasks including inpainting, uncropping, and colorization.

Palette-Image-to-Image-Diffusion-Models
Velocity · 7d
+1.2
★ / day
Trend
steady
star history

This repository provides an unofficial PyTorch implementation of the Palette paper, which uses conditional diffusion models for image-to-image translation. The implementation includes U-Net architecture with attention mechanisms adapted from guided diffusion, encodes gamma conditioning, and supports distributed training (DDP) with EMA. The project covers three main tasks: image inpainting on CelebaHQ and Places2 datasets, uncropping, and colorization, with evaluation metrics including FID and IS scores.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.