DL之CNN：基于CRNN_OCR算法(keras,CNN+RNN)利用數(shù)據(jù)集(torch,mdb格式)訓(xùn)練來實(shí)現(xiàn)新圖片上不定長度字符串進(jìn)行識別—訓(xùn)練過程

?

?

目錄

輸出結(jié)果

設(shè)計(jì)思路

核心代碼

---

?

?

?

輸出結(jié)果

?

?

設(shè)計(jì)思路

?

?

?

?

核心代碼

from keras.callbacks import ModelCheckpoint, ReduceLROnPlateaunclass = len(characters) + 1 model, basemodel = get_model_train(height=imgH, nclass=nclass) import osif os.path.exists('./models/pretrain_models/keras.hdf5'):basemodel.load_weights('./models/pretrain_models/keras.hdf5')checkpointer = ModelCheckpoint(filepath="./models/ intermediate_model/model{epoch:02d}-{val_loss:.4f}.hdf5", monitor='val_loss',verbose=0, save_weights_only=False, save_best_only=True)rlu = ReduceLROnPlateau(monitor='val_loss', factor=0.1, patience=1, verbose=0, mode='auto', epsilon=0.0001,cooldown=0, min_lr=0)model.fit_generator(gen(train_loader, flag='train'), steps_per_epoch=1024,epochs=10000,validation_data=gen(test_loader, flag='test'),callbacks=[checkpointer, rlu],validation_steps=1024) #保存兩個h5文件 model.save_weights('./models/final_model/final_model_weights.h5') model.save('./models/final_model/final_model.h5')

?

?

?

?

?

總結(jié)

以上是生活随笔為你收集整理的DL之CNN：基于CRNN_OCR算法(keras,CNN+RNN)利用数据集(torch,mdb格式)训练来实现新图片上不定长度字符串进行识别—训练过程的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。