kornia
opencv-python
matplotlib