opencv-python
pillow
torch
torchvision