Refactor test_on_image.py

Add all necessary files to replicate training run

Add training scripts and initial model trained on 1% of the data.