Create image dataset for machine learning