pytorch神经网络训练（VGG-19）

VGG-19

1. 导包

import torch

import torch.nn as nn

import torch.optim as optim

import torchvision

from torchvision import datasets, transforms

from torch.utils.data import DataLoader

import matplotlib.pyplot as plt

1. 数据预处理和增强

transform = transforms.Compose([

transforms.Resize((224, 224)),  

调整图像大小为 224x224像素，符合VGG16输入

    

transforms.ToTensor(),

transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),  

])