torch>=1.7.0
torchvision>=0.5.0
numpy
scikit-learn>=0.22.2
pillow>=7.1
