providing the list of all generated images.

[vguerra]

all_image_names is not defined and it is needed for the train test split.

For the data collection use case, the image names can be simply generated from the number of images that were saved: npz_index.

Otherwise, running the Data Collection generation application will throw an error complaining that all_image_names is not defined.

[liampaull]

thanks for this.