numpy
matplotlib
pandas
pytorch-lightning
torch
torchvision
torchaudio
einops
opencv-python
scipy
pickle5
tsaug
