IndexError: too many indices for array while plotting predicted masks

Question

I am trying to plot my predictions but I am keep getting this error.When I reduce the index I get TypeError: Invalid shape (1,) for image data.

img_seg has shape (1,1)