告别单字切割！用PyTorch从零搭建CRNN，搞定不定长文本识别（附完整代码）

OCRCRNNPyTorch文本识别

|难度：普通|阅读约 15 分钟

从零构建CRNN：PyTorch实战不定长文本识别全解析

在计算机视觉领域，文本识别一直是个极具挑战性的任务。传统方法需要先对文本图像进行单字切割，再逐个识别，这种方法不仅效率低下，而且对复杂排版和变形文本的适应性较差。本文将带你用PyTorch从零实现一个端到端的CRNN模型，直接处理不定长文本识别，彻底告别繁琐的单字切割流程。

1. CRNN架构设计精要

CRNN（Convolutional Recurrent Neural Network）是当前最流行的不定长文本识别架构之一，它巧妙地将CNN的特征提取能力与RNN的序列建模能力结合起来。整个流程无需显式的字符分割，可以直接从整张图片预测文本序列。

1.1 网络结构全景图

一个标准的CRNN包含三个核心组件：

CNN特征提取器：将输入图像转换为特征序列
双向LSTM：捕捉序列的上下文依赖关系
CTC损失函数：解决序列对齐问题

python复制class CRNN(nn.Module):
    def __init__(self, imgH, nc, nclass, nh):
        super(CRNN, self).__init__()
        # CNN部分
        self.cnn = nn.Sequential(
            nn.Conv2d(nc, 64, 3, 1, 1), nn.ReLU(True),
            nn.MaxPool2d(2, 2),
            # 更多卷积层...
        )
        # RNN部分
        self.rnn = nn.Sequential(
            BidirectionalLSTM(512, nh, nh),
            BidirectionalLSTM(nh, nh, nclass)
        )

1.2 CNN设计的特殊考量

文本图像通常具有"高较小、宽较长"的特点，这要求我们在CNN设计中做出特殊调整：

池化策略：后两个池化层采用1×2的窗口而非传统的2×2，以保留更多水平方向的信息
特征图高度：最终特征图的高度必须压缩为1，这要求输入图像高度是16的倍数
感受野对齐：每个特征向量对应原始图像的一个矩形区域，这些区域保持从左到右的顺序

提示：输入图像尺寸建议设置为32像素高，宽度按比例缩放。例如对于280×32的输入，CNN输出特征序列长度（时间步数）为40。

需要模型API调用？免费领10W Token，多模型网关一键接入 Claude、DeepSeek 等主流模型。

2. 双向LSTM的序列建模

CNN输出的特征序列需要送入循环神经网络进行时序建模。这里我们使用双向LSTM来同时捕捉前后文信息。

2.1 双向LSTM实现细节

python复制class BidirectionalLSTM(nn.Module):
    def __init__(self, nIn, nHidden, nOut):
        super(BidirectionalLSTM, self).__init__()
        self.rnn = nn.LSTM(nIn, nHidden, bidirectional=True)
        self.embedding = nn.Linear(nHidden * 2, nOut)

    def forward(self, input):
        recurrent, _ = self.rnn(input)
        T, b, h = recurrent.size()
        t_rec = recurrent.view(T * b, h)
        output = self.embedding(t_rec)
        return output.view(T, b, -1)

关键参数说明：

参数	典型值	说明
nIn	512	输入特征维度
nHidden	256	LSTM隐藏单元数
nOut	字符类别数+1	输出维度（含blank类别）

2.2 序列到序列的映射

每个时间步，LSTM接收一个512维的特征向量，输出所有字符的概率分布。对于40个时间步的序列，我们得到一个40×(字符类别数+1)的后验概率矩阵。

3. CTC损失原理与实现

CTC（Connectionist Temporal Classification）是解决不定长序列对齐的关键，它允许模型在不需要精确字符位置标注的情况下进行训练。

3.1 CTC的核心机制

Blank机制：引入空白符号"-"处理重复字符
多路径合并：不同对齐路径可能对应相同标签
概率求和：计算所有有效路径的概率之和

python复制# PyTorch中的CTC损失计算
criterion = nn.CTCLoss()
loss = criterion(log_probs, targets, input_lengths, target_lengths)

3.2 训练与解码的差异

训练阶段：

计算所有可能路径的概率和
使用动态规划高效计算
目标是最大化正确标签的概率

测试阶段：

采用贪心搜索或束搜索(beam search)
取每个时间步最可能的输出
合并重复字符并移除blank

python复制# 解码示例
decoded, _ = torch.nn.functional.ctc_beam_search_decoder(
    log_probs, 
    seq_len, 
    beam_width=10
)

4. 完整训练流程实战

4.1 数据准备与增强

我们使用合成数据来训练CRNN模型，关键步骤包括：

文本生成：从语料库随机抽取句子
字体渲染：使用不同字体、大小、颜色
噪声添加：椒盐噪声、高斯模糊、透视变换
尺寸归一化：高度统一为32像素，宽度保持比例

python复制def generate_text_image(text, width=280, height=32):
    # 创建空白图像
    image = Image.new('L', (width, height), color=255)
    draw = ImageDraw.Draw(image)
    
    # 随机选择字体和大小
    font_size = random.randint(24, 32)
    font = ImageFont.truetype(random.choice(fonts), font_size)
    
    # 绘制文本
    draw.text((10, (height-font_size)//2), text, font=font, fill=0)
    
    # 添加噪声
    image = add_noise(np.array(image))
    return image

4.2 数据加载与批处理

由于文本长度不一，我们使用稀疏矩阵存储标签：

python复制def collate_fn(batch):
    images = torch.stack([item[0] for item in batch])
    # 标签转为稀疏表示
    targets = [item[1] for item in batch]
    target_lengths = torch.tensor([len(t) for t in targets])
    targets = torch.cat(targets)
    return images, targets, target_lengths

4.3 模型训练技巧

学习率调度：采用指数衰减学习率
梯度裁剪：防止RNN梯度爆炸
早停机制：基于验证集准确率

python复制optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
scheduler = torch.optim.lr_scheduler.ExponentialLR(optimizer, gamma=0.98)

for epoch in range(100):
    for images, targets, target_lengths in train_loader:
        # 前向传播
        log_probs = model(images)
        input_lengths = torch.full(
            (images.size(0),), 
            log_probs.size(0), 
            dtype=torch.long
        )
        
        # 计算CTC损失
        loss = criterion(log_probs, targets, input_lengths, target_lengths)
        
        # 反向传播
        optimizer.zero_grad()
        loss.backward()
        torch.nn.utils.clip_grad_norm_(model.parameters(), 5)
        optimizer.step()
    
    scheduler.step()

5. 模型优化与部署

5.1 性能提升技巧

残差连接：在CNN部分加入残差块
注意力机制：替代或辅助CTC
语言模型融合：后处理时结合n-gram统计

python复制class ResidualBlock(nn.Module):
    def __init__(self, in_channels, out_channels, stride=1):
        super().__init__()
        self.conv1 = nn.Conv2d(in_channels, out_channels, 3, stride, 1)
        self.bn1 = nn.BatchNorm2d(out_channels)
        self.conv2 = nn.Conv2d(out_channels, out_channels, 3, 1, 1)
        self.bn2 = nn.BatchNorm2d(out_channels)
        
        if stride != 1 or in_channels != out_channels:
            self.shortcut = nn.Sequential(
                nn.Conv2d(in_channels, out_channels, 1, stride),
                nn.BatchNorm2d(out_channels)
            )
        else:
            self.shortcut = nn.Identity()

    def forward(self, x):
        out = F.relu(self.bn1(self.conv1(x)))
        out = self.bn2(self.conv2(out))
        out += self.shortcut(x)
        return F.relu(out)

5.2 实际部署注意事项

输入归一化：保持与训练时一致的预处理
动态宽度：支持任意宽度输入
批处理优化：对相同长度的输入进行批处理
量化加速：使用FP16或INT8量化

在真实场景中测试模型时，我发现对模糊和低对比度文本的识别仍有提升空间。一个实用的技巧是在推理时对图像进行适度的锐化和对比度增强，这能显著提升困难样本的识别率。