torch
torchvision
pillow  # For PIL