torch
torchvision
opencv-contrib-python
lightning

[dev]
pytest
black
isort
build
