Add support for image augmentation

[obrienwr]

currently the only transformation we do is the resize to 224x224, for now we can simply hard-code a RandomRotate or something along those lines to increase the chance of this training well.