colorama
matplotlib
numpy
pillow-simd
torch
torchvision
tqdm
