12、卷积网络在计算机视觉中的应用与高级卷积类型

卷积网络在计算机视觉中的应用与高级卷积类型

图像增强技术

常见的图像增强技术包括：
- 缩放（Zoom in/out）
- 裁剪（Crop）
- 倾斜（Skew）
- 对比度和亮度调整（Contrast and brightness adjustment）

用 PyTorch 进行图像分类

步骤

1. 选择设备 ：优先选择 GPU，因为该神经网络比 MNIST 示例中的网络更大，使用 CPU 训练会非常缓慢。

import torch
from torchsummary import summary
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

2. 加载训练数据集 ：对训练数据进行随机水平和垂直翻转，并使用 z - score 归一化。

import torchvision.transforms as transforms
from torchvision import datasets
from torch.utils.data import DataLoader

# Training dataset
train_transform = transforms.Compose(