torch>=2.0.0
torchvision
torchmetrics
tensorboard
kornia
seaborn
albumentations
click
einops
pycocotools
scikit-image
scikit-learn
tqdm
timm
wandb
visdom

[test]
pytest
