PyTorch Lightning Image Classification Pipeline

A flexible image classification pipeline using PyTorch Lightning, supporting various models (CNN, VAE, DIFFUSION) and datasets.

Quick Start

1. Clone repo: git clone https://github.com/HansaemIanOh/vision.git && cd vision
2. Install: pip install -r requirements.txt
3. Prepare dataset in ImageFolder format
4. Run: python run.py

For detailed usage, configuration options, and customization, see the full documentation.