torch
opencv-python
matplotlib
transformers
tqdm
albumentations