Production training code for Machine Learning models TODO: add current training code for each image related machine learning model to this repository, with relevant documentation.