PyTorch深度学习实战——交通标志识别

PyTorch深度学习实战——交通标记识别

- 0. 前言
- 1. 交通标志识别
- - 1.1 数据集介绍
  - 1.2 数据增强和批归一化
- 3. 交通标志检测
- 相关链接

0. 前言

在道路交通场景中，交通标志识别作为驾驶辅助系统与无人驾驶车辆中不可缺少的技术，为车辆行驶中提供了安全保障。在道路上行驶的车辆，道路周围的环境包括许多重要的交通标志信息，根据交通标志信息在道路上做出正确的驾驶行为，通常能够避免发生交通事故。交通标志识别可以检测并识别当前行驶道路上的交通标志，然后得出有关道路的必要信息。
但交通标志会受到车辆的运动状态、光照以及遮挡等环境因素的影响，因此如何使车辆在道路交通中快速准确地帮助驾驶员识别交通标志已经成为智能交通领域的热点问题之一。鉴于交通标志识别在自动驾驶等应用中具有重要作用，在节中，我们将学习使用卷积神经网络实现交通标志识别。

1. 交通标志识别

1.1 数据集介绍

德国交通标志识别基准 (German Traffic Sign Recognition Benchmark, GTSRB) 是高级驾驶辅助系统和自动驾驶领域的交通标志图像分类基准。其中共包含 43 种不同类别的交通标志。可以在官方网页中下载相关数据集。
每张图片包含一个交通标志，图像包含实际交通标志周围的环境像素，大约为交通标志尺寸的 10% (至少为 5 个像素)，图像以 PPM 格式存储，图像尺寸在 15x15 到 250x250 像素之间。

1.2 数据增强和批归一化

在介绍神经网络时，我们已经了解了利用数据增强可以提高模型准确性。在现实世界中，我们会遇到具有不同特征的图像，例如，某些图像可能更亮，某些图像中的感兴趣对象可能在图像边缘附近，而某些图像可能较为模糊。在本节中，我们将介绍如何使用数据增强和批归一化提高模型的准确率。
为了了解数据增强和批归一化对模型性能的影响，我们将使用交通标志数据集训练交通标志识别模型，并评估以下三种情况：

不使用批归一化和数据增强
只使用批归一化，但不使用数据增强
同时使用批归一化和数据增强

除了以上不同外，数据集及其预处理方法完全相同。

3. 交通标志检测

首先考虑不使用批归一化和数据增强的情况，使用 PyTorch 实现交通标志识别。

(1) 下载数据集并导入相关库：

import os
import torch
import numpy as np
from torch.utils.data import DataLoader, Dataset
from glob import glob
from random import randint
import cv2
from pathlib import Path
import torch.nn as nn
from torch import optim
from matplotlib import pyplot as plt
import pandas as pddevice = 'cuda' if torch.cuda.is_available() else 'cpu'

(2) 指定与输出类别对应的索引：

from torchvision import transforms as T
classIds = pd.read_csv('signnames.csv')
classIds.set_index('ClassId', inplace=True)
classIds = classIds.to_dict()['SignName']
classIds = {f'{k:05d}':v for k,v in classIds.items()}
id2int = {v:ix for ix,(k,v) in enumerate(classIds.items())}

(3) 定义图像转换管道，执行图像转换操作(不使用数据增强)：

trn_tfms = T.Compose([T.ToPILImage(),T.Resize(32),T.CenterCrop(32),# T.ColorJitter(brightness=(0.8,1.2), # contrast=(0.8,1.2), # saturation=(0.8,1.2), # hue=0.25),# T.RandomAffine(5, translate=(0.01,0.1)),T.ToTensor(),T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])val_tfms = T.Compose([T.ToPILImage(),T.Resize(32),T.CenterCrop(32),T.ToTensor(),T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])

在以上代码中，对输入图像进行了一系列转换——首先将图像尺寸调整为 128 (最小边为 128），然后从图像中心进行裁剪。此外，使用 .ToTensor() 方法对图像进行缩放(使像素值位于 0 到 1 之间)，最后对图像进行归一化处理，以便使用预训练模型。
取消以上代码中的注释并重新运行即可执行数据增强。此外，我们并不会对 val_tfms 执行数据增强，因为在模型训练期间没有使用这些图像。但是，val_tfms 图像需要通过与 trn_tfms 相同的转换管道。

(4) 定义数据集类 GTSRB：

class GTSRB(Dataset):"""Road Sign Detection dataset."""def __init__(self, files, transform=None):self.files = filesself.transform = transformdef __len__(self):return len(self.files)def __getitem__(self, ix):fpath = self.files[ix]clss = os.path.basename(Path(fpath).parent)img = cv2.imread(fpath)img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)return img, classIds[clss]def choose(self):return self[randint(len(self))]def collate_fn(self, batch):imgs, classes = list(zip(*batch))if self.transform:imgs = [self.transform(img)[None] for img in imgs]classes = [torch.tensor([id2int[clss]]) for clss in classes]imgs, classes = [torch.cat(i).to(device) for i in [imgs, classes]]return imgs, classes

(5) 创建训练、验证数据集和数据加载器：

all_files = glob('GTSRB/Final_Training/Images/*/*.ppm')
np.random.shuffle(all_files)from sklearn.model_selection import train_test_split
trn_files, val_files = train_test_split(all_files, random_state=1)trn_ds = GTSRB(trn_files, transform=trn_tfms)
val_ds = GTSRB(val_files, transform=val_tfms)
trn_dl = DataLoader(trn_ds, 32, shuffle=True, collate_fn=trn_ds.collate_fn)
val_dl = DataLoader(val_ds, 32, shuffle=False, collate_fn=val_ds.collate_fn)

(6) 定义模型 SignClassifier：

def convBlock(ni, no):return nn.Sequential(nn.Dropout(0.2),nn.Conv2d(ni, no, kernel_size=3, padding=1),nn.ReLU(inplace=True),#nn.BatchNorm2d(no),nn.MaxPool2d(2),)class SignClassifier(nn.Module):def __init__(self):super().__init__()self.model = nn.Sequential(convBlock(3, 64),convBlock(64, 64),convBlock(64, 128),convBlock(128, 64),nn.Flatten(),nn.Linear(256, 256),nn.Dropout(0.2),nn.ReLU(inplace=True),nn.Linear(256, len(id2int)))self.loss_fn = nn.CrossEntropyLoss()def forward(self, x):return self.model(x)def compute_metrics(self, preds, targets):ce_loss = self.loss_fn(preds, targets)acc = (torch.max(preds, 1)[1] == targets).float().mean()return ce_loss, acc

当需要在模型中使用 BatchNormalization (批归一化)时，需要取消注释以上代码中注释行。

(7) 定义使用批数据对模型进行训练和验证的函数：

def train_batch(model, data, optimizer, criterion):ims, labels = data_preds = model(ims)optimizer.zero_grad()loss, acc = criterion(_preds, labels)loss.backward()optimizer.step()return loss.item(), acc.item()@torch.no_grad()
def validate_batch(model, data, criterion):ims, labels = data_preds = model(ims)loss, acc = criterion(_preds, labels)return loss.item(), acc.item()

(8) 定义模型并对其进行训练：

model = SignClassifier().to(device)
criterion = model.compute_metrics
optimizer = optim.Adam(model.parameters(), lr=1e-3)
n_epochs = 50train_loss_epochs_no_aug_no_bn = []
train_acc_epochs_no_aug_no_bn = []
val_loss_epochs_no_aug_no_bn = []
val_acc_epochs_no_aug_no_bn = []
for ex in range(n_epochs):train_loss = []train_acc = []val_loss = []val_acc = []N = len(trn_dl)for bx, data in enumerate(trn_dl):loss, acc = train_batch(model, data, optimizer, criterion)train_loss.append(loss)train_acc.append(acc)N = len(val_dl)for bx, data in enumerate(val_dl):loss, acc = validate_batch(model, data, criterion)val_loss.append(loss)val_acc.append(acc)train_loss_epochs_no_aug_no_bn.append(np.average(train_loss))train_acc_epochs_no_aug_no_bn.append(np.average(train_acc))val_loss_epochs_no_aug_no_bn.append(np.average(val_loss))val_acc_epochs_no_aug_no_bn.append(np.average(val_acc))if ex == 10:optimizer = optim.Adam(model.parameters(), lr=1e-4)epochs = np.arange(50)+1
import matplotlib.pyplot as plt
plt.subplot(121)
plt.plot(epochs, train_loss_epochs_no_aug_no_bn, 'bo', label='Training loss')
plt.plot(epochs, val_loss_epochs_no_aug_no_bn, 'r', label='Test loss')
plt.title('Training and Test loss over increasing epochs \n with no batchnormalization and augmentation')
plt.xlabel('Epochs')
plt.ylabel('Loss')
plt.legend()
plt.grid('off')
plt.subplot(122)
plt.plot(epochs, train_acc_epochs_no_aug_no_bn, 'bo', label='Training accuracy')
plt.plot(epochs, val_acc_epochs_no_aug_no_bn, 'r', label='Test accuracy')
plt.title('Training and Test accuracy over increasing epochs \n with no batchnormalization and augmentation')
plt.xlabel('Epochs')
plt.ylabel('Accuracy')
plt.legend()
plt.grid('off')
plt.show()