Skip to main content

You are not logged in. Your edit will be placed in a queue until it is peer reviewed.

We welcome edits that make the post easier to understand and more valuable for readers. Because community members review edits, please try to make the post substantially better than how you found it, for example, by fixing grammar or adding additional resources and hyperlinks.

Required fields*

4
  • 1
    Could you please share minimal reproducible example? Thx Commented Sep 1, 2024 at 19:08
  • @JohnnyCheesecutter I edited the question and added a link to the ipynb notebook. Commented Sep 2, 2024 at 5:55
  • The dataloader iterator produces a batch of images. Make sure you are plotting individual images and not the whole batch with plt.imshow(). Commented Sep 2, 2024 at 5:57
  • @berinaniesh thanks for the help, but I'm only taking the first image from the batch, not the whole batch Commented Sep 2, 2024 at 6:02