本發(fā)明涉及圖像識別,尤其涉及一種數(shù)字表計讀數(shù)的識別方法、裝置、電子設(shè)備及存儲介質(zhì)。
背景技術(shù):
1、數(shù)字表計作為一種傳統(tǒng)的測量工具,其構(gòu)造簡單、價格便宜、性能可靠且方便維護(hù),因此在各種工業(yè)生產(chǎn)和變電站中得到廣泛應(yīng)用。由于數(shù)字表計讀數(shù)通常通過人工來進(jìn)行讀取,容易受到主觀因素和環(huán)境因素的干擾從而導(dǎo)致數(shù)據(jù)不準(zhǔn)確。
2、目前現(xiàn)有技術(shù)中,對數(shù)字表計圖像中的讀數(shù)進(jìn)行識別時,通常采用圖像處理的方法進(jìn)行圖像分割,確定每個數(shù)字的位置后對每個數(shù)字進(jìn)行識別。在圖像分割過程中受到光照影響較大容易出現(xiàn)分割不準(zhǔn)確和數(shù)字不完整等問題,從而影響數(shù)字識別的準(zhǔn)確性。當(dāng)采用劃分矩形小塊區(qū)域進(jìn)行全局搜索小區(qū)域是否存在數(shù)字進(jìn)行圖像分割時,受到數(shù)字模型準(zhǔn)確度和圖像數(shù)字區(qū)域傾斜等影響,也將導(dǎo)致數(shù)字識別不準(zhǔn)確。因此,目前亟需一種在數(shù)字區(qū)域傾斜下也能夠準(zhǔn)確識別數(shù)字的方法。
技術(shù)實現(xiàn)思路
1、有鑒于此,本發(fā)明的目的是為了克服現(xiàn)有技術(shù)中的不足,提供一種數(shù)字表計讀數(shù)的識別方法、裝置、電子設(shè)備及存儲介質(zhì)。
2、本發(fā)明提供如下技術(shù)方案:
3、第一方面,本公開實施例中提供了一種數(shù)字表計讀數(shù)的識別方法,所述方法包括:
4、獲取數(shù)字表計的初始建模圖和測試圖,對所述初始建模圖進(jìn)行第一矯正處理得到矯正建模圖;
5、基于yolov8網(wǎng)絡(luò)模型判斷所述矯正建模圖和所述測試圖是否能夠識別數(shù)字表盤區(qū)域,若能識別所述數(shù)字表盤區(qū)域,則基于目標(biāo)檢測法計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量,若不能識別所述數(shù)字表盤區(qū)域,則基于特征點匹配法計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量;
6、根據(jù)所述預(yù)置位偏移量對所述測試圖進(jìn)行第二矯正處理得到最終測試圖;
7、基于resnet34網(wǎng)絡(luò)模型對所述最終測試圖進(jìn)行識別,得到數(shù)字表計讀數(shù)。
8、根據(jù)本技術(shù)公開的一種具體實施方式,所述獲取數(shù)字表計的初始建模圖和測試圖,對所述初始建模圖進(jìn)行第一矯正處理得到矯正建模圖的步驟,包括:
9、判斷所述初始建模圖的數(shù)字區(qū)域是否傾斜;
10、若傾斜,則標(biāo)定所述數(shù)字區(qū)域中傾斜線的矯正點,根據(jù)所述矯正點對所述初始建模圖進(jìn)行第一矯正處理;
11、若不傾斜,則對所述數(shù)字區(qū)域按照預(yù)設(shè)標(biāo)定方式進(jìn)行標(biāo)定。
12、根據(jù)本技術(shù)公開的一種具體實施方式,所述基于yolov8網(wǎng)絡(luò)模型判斷所述矯正建模圖和所述測試圖是否能夠識別數(shù)字表盤區(qū)域的步驟,包括:
13、對所述yolov8模型進(jìn)行加載預(yù)訓(xùn)練,得到預(yù)訓(xùn)練yolov8模型;
14、采集數(shù)字表計圖像,對所述數(shù)字表計圖像進(jìn)行目標(biāo)標(biāo)注得到數(shù)據(jù)集;
15、通過所述數(shù)據(jù)集對所述預(yù)訓(xùn)練yolov8模型進(jìn)行訓(xùn)練,得到目標(biāo)區(qū)域識別模型;
16、將所述矯正建模圖和所述測試圖輸入至所述目標(biāo)區(qū)域識別模型,輸出目標(biāo)區(qū)域。
17、根據(jù)本技術(shù)公開的一種具體實施方式,所述若能識別所述數(shù)字表盤區(qū)域,則基于目標(biāo)檢測法計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量的步驟,包括:
18、判斷所述矯正建模圖的寬與高比值與所述測試圖的寬與高比值的差值是否在預(yù)設(shè)差值范圍內(nèi),若所述差值在所述預(yù)設(shè)差值范圍以上,則基于特征點匹配法計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量;
19、若所述差值小于所述預(yù)設(shè)差值范圍,則確定所述矯正建模圖和所述測試圖的矩形框中心點,根據(jù)所述矩形框中心點計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量。
20、根據(jù)本技術(shù)公開的一種具體實施方式,所述若不能識別所述數(shù)字表盤區(qū)域,則基于特征點匹配法計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量的步驟,包括:
21、對所述矯正建模圖和所述測試圖進(jìn)行預(yù)處理得到所述矯正建模圖和所述測試圖的特征點;
22、對所述特征點進(jìn)行特征點匹配處理,得到匹配特征點;
23、對所述匹配特征點進(jìn)行偏移量計算得到偏移值;
24、對所述偏移值進(jìn)行聚類處理,得到所述矯正建模圖和所述測試圖的預(yù)置位偏移量。
25、根據(jù)本技術(shù)公開的一種具體實施方式,所述根據(jù)所述預(yù)置位偏移量對所述測試圖進(jìn)行第二矯正處理得到最終測試圖的步驟,包括:
26、判斷所述預(yù)置位偏移量是否在預(yù)設(shè)偏移范圍內(nèi),若所述預(yù)置位偏移量在所述預(yù)設(shè)偏移范圍以上,則將所述預(yù)置位偏移量重置為0并判斷是否為首次計算所述預(yù)置位偏移量,若所述預(yù)置位偏移量小于所述預(yù)設(shè)偏移范圍,則判斷是否為首次計算所述預(yù)置位偏移量;
27、若是首次計算所述預(yù)置位偏移量,則判斷所述矯正建模圖中數(shù)字區(qū)域是否傾斜,若所述矯正建模圖中數(shù)字區(qū)域傾斜,則對所述矯正建模圖和所述測試圖進(jìn)行第一矯正處理并返回至基于yolov8網(wǎng)絡(luò)模型判斷所述矯正建模圖和所述測試圖是否能夠識別數(shù)字表盤區(qū)域的步驟,若所述矯正建模圖中數(shù)字區(qū)域未傾斜,則對所述測試圖進(jìn)行第二矯正處理得到最終測試圖;
28、若不是首次計算所述預(yù)置位偏移量,則對所述測試圖進(jìn)行第二矯正處理得到最終測試圖。
29、根據(jù)本技術(shù)公開的一種具體實施方式,所述基于resnet34網(wǎng)絡(luò)模型對所述最終測試圖進(jìn)行識別,得到數(shù)字表計讀數(shù)的步驟,包括:
30、定義所述resnet34網(wǎng)絡(luò)模型的網(wǎng)絡(luò)結(jié)構(gòu),其中,所述網(wǎng)絡(luò)結(jié)構(gòu)包括卷積層、池化層和殘差塊;
31、設(shè)置所述resnet34網(wǎng)絡(luò)模型的訓(xùn)練參數(shù),其中,所述訓(xùn)練參數(shù)包括學(xué)習(xí)率、優(yōu)化器、損失函數(shù)、訓(xùn)練輪次和批量大?。?/p>
32、構(gòu)建數(shù)據(jù)集,通過所述數(shù)據(jù)集對所述resnet34網(wǎng)絡(luò)模型進(jìn)行訓(xùn)練,得到讀數(shù)識別模型;
33、將所述最終測試圖輸入至所述讀數(shù)識別模型,輸出數(shù)字表計讀數(shù)。
34、第二方面,本公開實施例中提供了一種數(shù)字表計讀數(shù)的識別裝置,所述裝置包括:
35、獲取模塊,用于獲取數(shù)字表計的初始建模圖和測試圖,對所述初始建模圖進(jìn)行第一矯正處理得到矯正建模圖;
36、計算模塊,用于基于yolov8網(wǎng)絡(luò)模型判斷所述矯正建模圖和所述測試圖是否能夠識別數(shù)字表盤區(qū)域,若能識別所述數(shù)字表盤區(qū)域,則基于目標(biāo)檢測法計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量,若不能識別所述數(shù)字表盤區(qū)域,則基于特征點匹配法計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量;
37、矯正模塊,用于根據(jù)所述預(yù)置位偏移量對所述測試圖進(jìn)行第二矯正處理得到最終測試圖;
38、識別模塊,用于基于resnet34網(wǎng)絡(luò)模型對所述最終測試圖進(jìn)行識別,得到數(shù)字表計讀數(shù)。
39、第三方面,本公開實施例中提供了一種電子設(shè)備,所述電子設(shè)備包括存儲器和處理器,所述存儲器存儲有計算機(jī)程序,所述處理器執(zhí)行所述計算機(jī)程序時實現(xiàn)第一方面中任一項所述的數(shù)字表計讀數(shù)的識別方法的步驟。
40、第四方面,本公開實施例中提供了一種計算機(jī)可讀存儲介質(zhì),所述計算機(jī)可讀存儲介質(zhì)存儲有計算機(jī)程序,所述計算機(jī)程序被處理器執(zhí)行時實現(xiàn)第一方面中任一項所述的數(shù)字表計讀數(shù)的識別方法的步驟。
41、上述本技術(shù)提供的數(shù)字表計讀數(shù)的識別方法,通過獲取數(shù)字表計的初始建模圖和測試圖,對所述初始建模圖進(jìn)行第一矯正處理得到矯正建模圖,基于yolov8網(wǎng)絡(luò)模型判斷所述矯正建模圖和所述測試圖是否能夠識別數(shù)字表盤區(qū)域,若能識別所述數(shù)字表盤區(qū)域,則基于目標(biāo)檢測法計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量,若不能識別所述數(shù)字表盤區(qū)域,則基于特征點匹配法計算所述矯正建模圖和所述測試圖的預(yù)置位偏移量,根據(jù)所述預(yù)置位偏移量對所述測試圖進(jìn)行第二矯正處理得到最終測試圖,基于resnet34網(wǎng)絡(luò)模型對所述最終測試圖進(jìn)行識別,得到數(shù)字表計讀數(shù)?;谀繕?biāo)檢測法和特征點匹配法對偏移量進(jìn)行計算并對數(shù)字區(qū)域進(jìn)行校準(zhǔn),能夠提高識別數(shù)字表計讀數(shù)的準(zhǔn)確率,提高了工作效率。
42、為使本發(fā)明的上述目的、特征和優(yōu)點能更明顯和易懂,下文特舉較佳實施例,并配合所附附圖,做詳細(xì)說明如下。