Efficient image to array training dataset in Python

Question

I\'m training a network that takes an image as an input and outputs a smaller 2D array (say 10x10) of scores for parts of the image.

I currently build the training da