Merge pull request #124 from mark-hoffmann/patch-1

Plots update

Author: jph00

deeplearning1/nbs/utils.py

@@ -82,8 +82,9 @@ def plots(ims, figsize=(12,6), rows=1, interp=False, titles=None):
         if (ims.shape[-1] != 3):
             ims = ims.transpose((0,2,3,1))
     f = plt.figure(figsize=figsize)
+    cols = len(ims)//rows if len(ims) % 2 == 0 else len(ims)//rows + 1
     for i in range(len(ims)):
-        sp = f.add_subplot(rows, len(ims)//rows, i+1)
+        sp = f.add_subplot(rows, cols, i+1)
         sp.axis('Off')
         if titles is not None:
             sp.set_title(titles[i], fontsize=16)