click
more-itertools
torchvision
torch
pillow
numpy
