Implement training and testing dataset generation for experiment01_a now that masks have been produced.