Pix2Pix Lightning

A lightweight implementation of the Pix2Pix image-to-image translation model using PyTorch Lightning.

• ✅ Modular codebase
• 🧪 Supports training and inference via YAML config files
• 🔬 Includes logging with Weights & Biases
• 📊 Optional evaluation with PSNR and SSIM

Getting Started

1. Install uv

curl -Ls https://astral.sh/uv/install.sh | sh

Or follow instructions: https://github.com/astral-sh/uv

2. Create a virtual environment and install dependencies

uv venv
uv sync

3. Train the model

uv run main.py