torch
opencv-python==4.5.4.60
transformers
albumentations
timm
tqdm