numpy
opencv-python
pytorch_lightning
torch
torchvision
transformers
seaborn
matplotlib