torch>=2.0
torchvision
compressai>=1.2
kornia>=0.6
piq>=0.7
pytorch-wavelets>=1.3
torchmetrics>=1.0
numpy
pillow
scikit-image
matplotlib
pandas
gdown
kagglehub

[dev]
pytest
black
ruff
