利用CRNN搭建文字识别模型,案例代码说明注释
时间: 2024-03-20 22:44:48 浏览: 33
以下是一个利用CRNN搭建文字识别模型的案例代码说明注释:
```python
# 导入必要的库
import os
import cv2
import numpy as np
from keras.models import Sequential
from keras.layers.convolutional import Conv2D, MaxPooling2D
from keras.layers.recurrent import GRU
from keras.layers.core import Dense, Activation, Reshape
from keras.optimizers import Adam
from keras.callbacks import EarlyStopping
from keras.backend import ctc_batch_cost, ctc_decode
# 设置参数
img_h = 32 # 图像高度
img_w = 280 # 图像宽度
batch_size = 64 # 批次大小
max_text_len = 20 # 最大文本长度
epochs = 50 # 训练迭代次数
num_classes = 37 # 类别数,包括26个英文字母、10个数字和1个空格
# 定义CRNN模型
def crnn():
# 定义序列模型
model = Sequential()
# 添加卷积层
model.add(Conv2D(64, (3, 3), padding='same', input_shape=(img_h, img_w, 1), activation='relu'))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(128, (3, 3), padding='same', activation='relu'))
model.add(Conv2D(128, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(256, (3, 3), padding='same', activation='relu'))
model.add(Conv2D(256, (3, 3), activation='relu'))
model.add(Conv2D(256, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(1, 2)))
model.add(Conv2D(512, (3, 3), padding='same', activation='relu'))
model.add(Conv2D(512, (3, 3), activation='relu'))
model.add(Conv2D(512, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(1, 2)))
model.add(Conv2D(512, (2, 2), padding='same', activation='relu'))
model.add(Conv2D(512, (2, 2), activation='relu'))
# 将输出变成序列形式
model.add(Reshape(target_shape=((img_w // 32), -1)))
# 添加循环层
model.add(GRU(256, return_sequences=True))
model.add(GRU(256, return_sequences=True))
# 添加全连接层
model.add(Dense(num_classes))
# 添加激活函数
model.add(Activation('softmax'))
# 返回模型
return model
# 定义CTC损失函数
def ctc_loss(y_true, y_pred):
return ctc_batch_cost(y_true, y_pred, np.ones(batch_size)*int(img_w/32), np.ones(batch_size)*max_text_len)
# 定义CTC解码函数
def ctc_decode_func(y_pred):
return ctc_decode(y_pred, np.ones(batch_size)*int(img_w/32), greedy=True, beam_width=100, top_paths=1)[0][0]
# 加载数据集
def load_data():
# 加载图像和标签
img_files = os.listdir('data/img')
X = []
y = []
for img_file in img_files:
img = cv2.imread('data/img/'+img_file, 0)
img = cv2.resize(img, (img_w, img_h))
X.append(img)
with open('data/label/'+img_file.split('.')[0]+'.txt', 'r', encoding='utf8') as f:
label = f.read().strip()
y.append(label)
# 将图像和标签转换为数组形式
X = np.array(X).reshape(-1, img_h, img_w, 1)
y = np.array(y)
# 将标签转换为one-hot编码形式
y = [np.array([char_to_idx[c] for c in text]) for text in y]
y = np.array(y)
y = np.zeros((y.shape[0], max_text_len, num_classes))
for i, t in enumerate(y_train):
for j, char_idx in enumerate(t):
y[i, j, char_idx] = 1
# 返回数据集
return X, y
# 训练模型
def train():
# 加载数据集
X_train, y_train = load_data()
# 定义模型
model = crnn()
# 编译模型
model.compile(loss=ctc_loss, optimizer=Adam())
# 设置EarlyStopping
early_stopping = EarlyStopping(monitor='val_loss', patience=5)
# 训练模型
model.fit(X_train, y_train, batch_size=batch_size, epochs=epochs, validation_split=0.2, callbacks=[early_stopping])
# 保存模型
model.save('model.h5')
# 测试模型
def test():
# 加载测试集
X_test, y_test = load_data()
# 加载模型
model = crnn()
model.load_weights('model.h5')
# 对测试集进行预测
y_pred = model.predict(X_test)
# 对预测结果进行解码
decoded = ctc_decode_func(y_pred)
# 计算准确率、召回率和F1值等指标
num_correct = 0
num_total = len(y_test)
for i in range(num_total):
if y_test[i] == decoded[i]:
num_correct += 1
accuracy = num_correct / num_total
print('准确率:', accuracy)
# 字符集
charset = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ '
# 字符到索引的映射表
char_to_idx = {char: idx for idx, char in enumerate(charset)}
# 索引到字符的映射表
idx_to_char = {idx: char for idx, char in enumerate(charset)}
```
这个案例包括以下步骤:
1. 定义CRNN模型,包括卷积神经网络(CNN)、循环神经网络(RNN)和全连接层。
2. 定义CTC损失函数和CTC解码函数。
3. 加载数据集,并将图像和标签转换为数组形式。
4. 训练模型,并在训练过程中使用EarlyStopping防止过拟合。
5. 测试模型,并计算准确率、召回率和F1值等指标。
在这个案例中,我们使用了Synthetic Chinese String Dataset数据集,它包括汉字、数字和英文字母的文本图像。在训练过程中,我们使用了CTC损失函数对序列输出进行训练,并使用Adam优化器进行模型优化。在测试过程中,我们使用CTC解码函数对预测结果进行解码,并计算准确率、召回率和F1值等指标。