python>=3.10.4
torch>=1.12.1
torchvision>=0.13.1
torchmetrics>=0.6.0
opencv-python>=4.5.5.64
pytorch-lightning>=1.4.2
omegaconf>=2.1.1
einops>=0.3.0
kornia>=0.6
aiofiles
gradio
# datasets[vision]>=2.4.0