The number of input images changes for every sample?
I.e. sometimes the current sample has only 1 input image and other times it has 3?
Usually all classes of a segmentation are saved in one segmentation map (similar to an image), so that we could possibly return just one segmentation target containing all 10 classes.