torch
opencv-python
torchvision