If you are generating the images, I guess I don’t understand why you just label them automatically during generation? That’s the neatest way to get a function that maps you input space to your desired outputs.