torch>=1.13.1
torchvision
Pillow
numpy
opencv-python
