torch
numpy
colorama
pillow