Convert dataset For converting data into png file instead of numpy array, running $ python preprocess_data.py The train data and test data and their corresponding annotation files would be in the './data/ folder. Start training Just running $ python train.py