Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers)

Cast the image to np.uint8 after scaling [0, 255] range will dismiss this warning. It seems like a feature in matplotlib, as discussed in this issue.

plt.imshow((out * 255).astype(np.uint8))