torch
torchvision
opencv-python
kornia
matplotlib
