torch
torchvision
Pillow
opencv-python