plot images in grid

PHOTO EMBED

Wed Feb 09 2022 18:13:34 GMT+0000 (UTC)

Saved by @somuSan #boilerplate #python #grid

# Visualize one training batch
for images, labels in train_data_loader:
    fig, ax = plt.subplots(figsize = (10, 10))
    ax.set_xticks([])
    ax.set_yticks([])
    ax.imshow(make_grid(images, 4).permute(1,2,0))
    break
content_copyCOPY

https://www.kaggle.com/jainamshah17/pytorch-starter-image-classification