pytorch_lightning
torch
torchvision
torchmetrics
opencv-python
albumentations

[dev]
pytest
tox
