Pytorch -- Fashion MNIST using feedforward neural networks

最新推荐文章于 2021-03-07 19:18:38 发布

jerry173985

最新推荐文章于 2021-03-07 19:18:38 发布

阅读量226

点赞数

分类专栏： Pytorch

本文链接：https://blog.csdn.net/jerry173985/article/details/110621284

版权

本文介绍了使用PyTorch对Fashion MNIST数据集进行图像分类的方法，包括数据预处理、构建全连接神经网络模型、利用GPU加速训练、模型训练过程以及样本预测与结果保存。

摘要由CSDN通过智能技术生成

Classifying images from Fashion MNIST using feedforward neural networks

import torch
import torchvision
import numpy as np
import matplotlib.pyplot as plt
import torch.nn as nn
import torch.nn.functional as F
from torchvision.datasets import FashionMNIST
from torchvision.transforms import ToTensor
from torchvision.utils import make_grid
from torch.utils.data.dataloader import DataLoader
from torch.utils.data import random_split
%matplotlib inline

project_name='fashion-feedforward'

Preparing the Data

dataset = FashionMNIST(root='data/', download=True, transform=ToTensor())
test_dataset = FashionMNIST(root='data/', train=False, transform=ToTensor())

val_size = 10000
train_size = len(dataset) - val_size
train_ds, val_ds = random_split(dataset, [train_size, val_size])
len(train_ds), len(val_ds)

(50000, 10000)

batch_size=128

train_loader = DataLoader(train_ds, batch_size, shuffle=True, num_workers=4, pin_memory=True)
val_loader = DataLoader(val_ds, batch_size*2, num_workers=4, pin_memory=True)
test_loader = DataLoader(test_dataset, batch_size*2, num_workers=4, pin_memory=True)

for images, labels in train_loader:
    print('images.shape:', images.shape)
    plt.figure(figsize=(16,8))
    plt.axis('off')
    plt.imshow(make_grid(images, nrow=16).permute((1, 2, 0)))