numpy
opencv-python
pandas
safetensors
torch
tqdm
wandb
