kornia
opencv-python
torch