albumentations matplotlib numpy pandas torch efficientnet-pytorch Pillow scikit-learn tqdm torchvision