torch
torchvision
torchaudio
numpy
opencv-contrib-python
tqdm

[dev]
black
pytest
