torch
numpy
Pillow
opencv-python
tqdm
tensorboard
pyyaml
safetensors