numpy
Pillow
opencv-python
torch >= 1.0.0
torchvision