CGSCORE / examples /image /testprint_mnist.py
Yaning1001's picture
Add files using upload-large-folder tool
fb9bb41 verified
import matplotlib.pyplot as plt
import torch
from torchvision import datasets
import numpy as py
dataset=datasets.MNIST('../data')
x=dataset.data[8888]
print(x.shape)
#print("pic info",x)
#x.save("imageexample.png")