mirror of
https://github.com/luzhisheng/js_reverse.git
synced 2025-04-23 05:39:22 +08:00
19 lines
473 B
Python
19 lines
473 B
Python
from torchvision.datasets import MNIST
|
|
from torchvision import transforms
|
|
from torch.utils.data import DataLoader
|
|
|
|
# 图片标准化处理目的就是提高识别准确度
|
|
my_transforms = transforms.Compose(
|
|
[
|
|
transforms.PILToTensor(),
|
|
]
|
|
)
|
|
|
|
mnist_train = MNIST(root="./MNIST_data", train=True, download=True, transform=my_transforms)
|
|
|
|
a = DataLoader(mnist_train, batch_size=1, shuffle=True)
|
|
for img, labels in a:
|
|
print(img)
|
|
print(labels)
|
|
exit()
|