torch
torchvision
numpy
scikit-learn
opencv-python
Pillow
efficientnet-pytorch