為何要調整模型呢?在前次的結果中可以看到辨識的結果對於角度較爲刁鑽或是不清楚的地方,在經過調整後是否能出現改善,以下是調整的重點。
在這邊筆者調整了訓練的參數,以下為程式以及為何要做這樣的修正的說明。
model.train(
data = r'C:\Users\使用者\Documents\projects\Car_License\dataset.yaml',
epochs = 350,
batch = 32, # 更改為32,增大批次大小
device = 'cuda',
imgsz = 320,
cache = True,
lr = 0.01, #修改學習率
augment = True #開啟資料增強
)
以下為訓練的結果
以整體的模型表現來說可以做出以下結論。
綜合來說,可以看到整體的數據都有提升的跡象,意味著調整上面的參數確實對結果有所幫助,那下一篇將開始介紹EasyOCR,接著會利用EasyOCR來進行辨識圖片中的文字,那我們下一篇見。