torch
torchvision
pillow
opencv-python
numpy