updates

2019-07-20 19:27:27 +02:00
parent a6877daa41
commit 05139a6bdd
2 changed files with 8 additions and 13 deletions
@@ -628,6 +628,7 @@ def plot_images(imgs, targets, paths=None, fname='images.jpg'):

    fig = plt.figure(figsize=(10, 10))
    bs, _, h, w = imgs.shape  # batch size, _, height, width
+    bs = min(bs, 16)  # limit plot to 16 images
    ns = np.ceil(bs ** 0.5)  # number of subplots

    for i in range(bs):