Imported from https://github.com/laramaktub/plant-classification-tf-train