Pytorch: torchvision.utils.make_grid函数的说明

原创

已于 2023-05-01 16:19:30 修改 · 1.6k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python #numpy #深度学习

于 2023-05-01 11:13:56 首次发布

Pytorch: torchvision.utils.make_grid函数的说明

网格化显示数据

# 环境准备
import numpy as np  # numpy数组库
import matplotlib.pyplot as plt  # 画图库
import torchvision.datasets as dataset  # 公开数据集的下载和管理
import torchvision.transforms as transforms  # 公开数据集的预处理库,格式转换
import torchvision
import torch.utils.data as data_utils  # 对数据集进行分批加载的工具集

# 2-1 准备数据集
train_data = dataset.MNIST(root="data",
                           train=True,
                           transform=transforms.ToTensor(),
                           download=True)

# 2-1 准备数据集
test_data = dataset.MNIST(root="data",
                          train=False,
                          transform=transforms.ToTensor(),
                          download=True)
# 批量数据读取
train_loader = data_utils.DataLoader(dataset=train_data,
                                     batch_size=64,
                                     shuffle=True)

test_loader = data_utils.DataLoader(dataset=test_da