torch
torchvision
PIL
numpy
argparse
copy