当前位置：首页 > news >正文

ResNet18手写数字识别：新手5分钟教程，无需任何基础

news 2026/8/14 0:47:45

ResNet18手写数字识别：新手5分钟教程，无需任何基础

1. 为什么选择ResNet18入门AI

中学生小明最近参加了学校AI兴趣班，但第一个项目就让他想放弃——老师给的代码在自己电脑完全跑不起来。这其实是很多AI新手的共同困境：环境配置复杂、代码依赖多、硬件要求高。而今天我要介绍的ResNet18手写数字识别项目，就是专为绝对零基础设计的入门方案。

ResNet18是深度学习领域最经典的图像分类模型之一，它的优势在于：

结构简单：只有18层神经网络，比动辄上百层的模型更轻量
效果稳定：在MNIST手写数字数据集上准确率可达99%以上
资源友好：不需要高端显卡，普通GPU甚至CPU都能运行

更重要的是，我们将使用预置好的镜像环境，完全跳过复杂的安装配置步骤。就像使用手机APP一样简单：点击→运行→出结果。

2. 环境准备：1分钟搞定

传统AI开发最劝退的就是环境配置，需要安装Python、PyTorch、CUDA等一堆工具，版本还要严格匹配。而我们将使用预配置好的PyTorch镜像，所有环境都已打包好，真正做到开箱即用。

2.1 获取镜像资源

在CSDN算力平台搜索"PyTorch基础镜像"，选择包含以下配置的版本：

Python 3.8+
PyTorch 1.12+
torchvision
CUDA 11.3（如果有GPU）

点击"一键部署"按钮，等待约30秒即可完成环境准备。这个过程就像下载一个APP，但省去了所有安装步骤。

2.2 验证环境

部署完成后，新建一个Jupyter Notebook，运行以下代码检查环境：

import torch print("PyTorch版本:", torch.__version__) print("GPU可用:", torch.cuda.is_available())

如果看到类似这样的输出，说明环境就绪：

PyTorch版本: 1.12.1 GPU可用: True

3. 手写数字识别实战

现在进入最核心的部分——用ResNet18识别手写数字。整个过程就像搭积木，我们只需要把现成的模块组合起来。

3.1 准备数据集

MNIST数据集包含6万张手写数字图片，每张都是28x28的灰度图。PyTorch已经内置了这个数据集，只需几行代码就能下载：

from torchvision import datasets, transforms # 定义数据转换（标准化） transform = transforms.Compose([ transforms.Resize(32), # ResNet18需要32x32输入 transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,)) ]) # 下载数据集 train_data = datasets.MNIST( root='data', train=True, download=True, transform=transform ) test_data = datasets.MNIST( root='data', train=False, download=True, transform=transform )

3.2 加载ResNet18模型

虽然ResNet18原本是为ImageNet（224x224彩色图）设计的，但我们可以稍作修改来适应MNIST：

import torch.nn as nn from torchvision import models # 加载预训练模型 model = models.resnet18(pretrained=False) # 修改第一层卷积（原始是3通道输入，我们改为1通道） model.conv1 = nn.Conv2d(1, 64, kernel_size=7, stride=2, padding=3, bias=False) # 修改最后一层全连接（原始是1000类输出，我们改为10类） num_features = model.fc.in_features model.fc = nn.Linear(num_features, 10)

3.3 训练模型

下面是训练代码，我已经添加了详细注释：

from torch.utils.data import DataLoader import torch.optim as optim # 创建数据加载器 train_loader = DataLoader(train_data, batch_size=64, shuffle=True) test_loader = DataLoader(test_data, batch_size=64, shuffle=False) # 定义损失函数和优化器 criterion = nn.CrossEntropyLoss() optimizer = optim.SGD(model.parameters(), lr=0.01, momentum=0.9) # 训练5个epoch（完整遍历数据集5次） for epoch in range(5): model.train() # 设置为训练模式 for images, labels in train_loader: # 前向传播 outputs = model(images) loss = criterion(outputs, labels) # 反向传播 optimizer.zero_grad() loss.backward() optimizer.step() # 每个epoch结束后测试准确率 model.eval() # 设置为评估模式 correct = 0 total = 0 with torch.no_grad(): for images, labels in test_loader: outputs = model(images) _, predicted = torch.max(outputs.data, 1) total += labels.size(0) correct += (predicted == labels).sum().item() print(f'Epoch [{epoch+1}/5], 准确率: {100 * correct / total:.2f}%')

运行这段代码，你会看到类似这样的输出：

Epoch [1/5], 准确率: 98.12% Epoch [2/5], 准确率: 98.76% Epoch [3/5], 准确率: 99.01% Epoch [4/5], 准确率: 99.15% Epoch [5/5], 准确率: 99.23%

4. 测试你自己的手写数字

训练好的模型可以保存下来，随时用来识别新的手写数字：

import matplotlib.pyplot as plt from PIL import Image # 保存模型 torch.save(model.state_dict(), 'mnist_resnet18.pth') # 加载模型（如果重启了环境） model.load_state_dict(torch.load('mnist_resnet18.pth')) # 测试自己的图片 def predict_image(image_path): img = Image.open(image_path).convert('L') # 转为灰度图 img = transform(img).unsqueeze(0) # 添加batch维度 model.eval() with torch.no_grad(): output = model(img) _, predicted = torch.max(output, 1) return predicted.item() # 示例：识别数字7 result = predict_image('my_digit.png') print(f'识别结果为: {result}')

你可以用画图工具写一个数字，保存为my_digit.png试试看！