This one might be okay, hard to tell before it is working. But the label array should be the same size, shape, number of pixels as the input image.