torch>=1.5.0
torchvision
kornia<=0.4.1
tqdm
numpy
ipython
pillow
future
decorator
pytest
pytest-mock
coverage
coveralls
scikit-learn
