A modified version of fastai's show_images function that applies an appropriate colormap to 2D images

Dim_Imsshow.py

def Dim_Imsshow(ims, nrows=1, ncols=None, titles= None, **kwargs):
  "Show all images 'ims' as subplots with 'rows' using 'titles'."
  if ncols is None: ncols = int(math.ceil(len(ims)/nrows))
  if titles is None: titles = [None]*len(ims)
  axs = subplots(nrows, ncols, **kwargs)[1].flat
  for im,t,ax in zip(ims, titles, axs): 
    if im.ndim < 3:
      show_image(im, ax = ax, title= t, cmap='gray')
    else:
      show_image(im, ax = ax, title= t)