Browse Source

improve show() (#15)

In Colab, calling show() multiple times in one cell won't display the pictures unless these two functions are called.
pull/20/head
neverix 5 years ago
committed by GitHub
parent
commit
706a76cfea
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      rudalle/pipelines.py

+ 2
- 0
rudalle/pipelines.py View File

@ -94,3 +94,5 @@ def show(pil_images, nrow=4):
img = torchvision.transforms.functional.to_pil_image(img)
axs[0, i].imshow(np.asarray(img))
axs[0, i].set(xticklabels=[], yticklabels=[], xticks=[], yticks=[])
fix.show()
plt.show()

Loading…
Cancel
Save