efficientnet-pytorch
opencv-python
torch
pillow
matplotlib
gradio
torchvision
albumentations