train_data 

for images, labels in train_data:
    for images, labels in train_data:
        img = images[0]
        img = img.numpy()
        img = np.transpose(img, (1, 2, 0))
        plt.imshow(img)
        plt.show()
posted on 2021-05-03 15:24  Sleven  阅读(76)  评论(0)  编辑  收藏  举报