torch
Pillow
numpy
collections
matplotlib