kornia
torch
matplotlib
opencv-python
