opencv_python
kornia
torch
torchvision
matplotlib