torch
opencv-python
pyyaml

[dev]
pytest
black
isort
mypy
