torch
torchvision
scikit-learn
numpy
matplotlib
PIL
open-cv