opencv-python
matplotlib
git+https://github.com/kornia/kornia