numpy
opencv-python-headless
Pillow
psutil
pyyaml
torch
torchvision
tqdm
