import PIL.Image
from io import BytesIO   
from IPython.display import display, Image
import numpy as np

def show_img(a, fmt='png'):
  # https://stackoverflow.com/questions/19471814/display-multiple-images-in-one-ipython-notebook-cell
  a = np.uint8(a)
  f = BytesIO()
  PIL.Image.fromarray(a).save(f, fmt)
  display(Image(data=f.getvalue()))