torch
opencv-python
matplotlib
transformers
tqdm
albumentations
pandas
timm