《電子技術應用》
您所在的位置:首頁 > 模擬設計 > 設計應用 > 基于改進CenterNet的發票檢測算法
基于改進CenterNet的發票檢測算法
電子技術應用
萬成凱1,李居朋2
1.北京世紀瑞爾技術股份有限公司;2.北京交通大學 電子信息工程學院
摘要: 為了提高發票檢測準確性和效率,提出了一種基于CenterNet的發票檢測算法。首先,算法模型采用類似CSPDarkNet作為主干網絡,將Triplet Attention引入CSP結構中形成TA-CSP結構,主干網絡末端引入ASPP以提高網絡的感受野范圍,使模型能夠更好地理解圖像的上下文信息;其次,在網絡的Neck部分,采用CBAM來引導高低層特征融合,利用高層特征圖中語義信息對低層特征圖進行監督,以抑制低層特征圖中的背景噪聲;再次,在網絡的Head部分,算法在CenterNet網絡的基礎上增加4個通道的特征圖輸出,在發票檢測的同時實現發票朝向的預測;最后,在損失函數中增加朝向損失項,以解決發票朝向的優化。在測試數據集上的實驗結果表明,本文算法mAP優于CenterNet和YOLOv5s算法達到84.3%,有效提高了發票檢測準確率和魯棒性。
中圖分類號:TP391.41;U418.6 文獻標志碼:A DOI: 10.16157/j.issn.0258-7998.245560
中文引用格式: 萬成凱,李居朋. 基于改進CenterNet的發票檢測算法[J]. 電子技術應用,2025,51(6):71-78.
英文引用格式: Wan Chengkai,Li Jupeng. Detection algorithm for invoice based on improved CenterNet[J]. Application of Electronic Technique,2025,51(6):71-78.
Detection algorithm for invoice based on improved CenterNet
Wan Chengkai1,Li Jupeng2
1.Beijing Century Real Technology Co., Ltd.; 2.School of Electronic and Information Engineering, Beijing Jiaotong University
Abstract: In order to improve the accuracy and efficiency of invoice detection, a CenterNet based invoice detection algorithm is proposed. Firstly, the algorithm model adopts a backbone network similar to CSPDarkNet, introducing Triplet Attention into the CSP structure to form a TA-CSP structure, and introducing ASPP at the end of the backbone network to improve the receptive field range of the network, enabling the model to better understand the contextual information of the image; Secondly, in the Neck part of the network, CBAM is used to guide the fusion of high-level and low-level features, and the semantic information in high-level feature maps is used to supervise low-level feature maps to suppress background noise in low-level feature maps; Thirdly, in the Head section of the network, the algorithm adds four channels of feature map outputs based on the CenterNet network, achieving invoice orientation prediction while detecting invoices; Finally, an orientation loss term is added to the loss function to optimize the orientation of invoices. The experimental results on the test dataset show that the mAP of the proposed algorithm in this paper is superior to CenterNet and YOLOv5s algorithms reaching 84.3%, effectively improving the accuracy and robustness of invoice detection.
Key words : CenterNet;YOLO;object detection;CBAM;ASPP;Triplet Attention

引言

隨著社會的不斷發展,大量發票的錄入和歸檔,對于財務人員是相當繁重的工作。在以往的工作中,財務人員往往采用手工錄入的方式,這種錄入方式不但效率低下,而且常常因為工作人員的疲勞產生錯誤而造成損失。隨著圖像處理與深度學習技術的興起,越來越多的研究人員開始研究基于數字圖像技術的發票自動識別算法[1-2]。

基于數字圖像技術的發票自動識別通常包括發票檢測、發票信息區定位、字符定位、字符識別等步驟。其中首要的步驟就是發票檢測。發票檢測是檢測出一張圖像中是否存在發票,并對每張存在的發票進行精確定位。由于在實際的財務歸檔工作中發票的朝向往往上下左右各不相同,因此發票檢測不但要檢測出發票的類型、位置,還要同時檢測出發票的朝向。

目前基于深度學習的目標檢測方法可以分為one-stage和two-stage方法兩類。two-stage的方法如Faster R-CNN[3]。這類檢測方法整個檢測過程分為兩個階段。在第一個階段,算法需要找到一些可能的目標存在區域;在第二個階段,算法在這可能的區域上進行分類和位置回歸。這類方法檢測精度高,但運行速度通常會比較慢,難以滿足實時檢測的需求。

one-stage的方法如YOLO系列[4-8]、SSD[9]等。這類檢測方法是一個端到端的檢測過程,它可以直接回歸出物體的類別和位置。該類方法過程簡潔、檢測速度快,目前已被廣泛應用于各種目標檢測任務當中,但其準確性仍有待提高。

YOLO系列、SSD等檢測方法均屬于基于錨點(anchor)的方法,需要事先統計anchor尺寸和比例等先驗知識,而且在計算過程中,會計算大量無用的候選框。雖然算法后期可以通過非極大值抑制等方法去除多余的候選框,但會帶來計算開銷的增加。以CenterNet[10]為代表的無anchor檢測方法克服了基于anchor方法的缺點,可以直接對目標中心點和尺寸進行預測。

本文結合YOLOv5的主干網絡和CenterNet各自的優點,提出了一種改進的CenterNet發票檢測算法。算法模型采用參考了CSPDarkNet的主干網絡,引入了注意力機制,并且采用新的輸出結構和損失函數,可以端到端地檢測出發票的分類、位置和朝向。


本文詳細內容請下載:

http://www.jysgc.com/resource/share/2000006565


作者信息:

萬成凱1,李居朋2

(1.北京世紀瑞爾技術股份有限公司 北京 100085;

2.北京交通大學 電子信息工程學院,北京 100044)


Magazine.Subscription.jpg

此內容為AET網站原創,未經授權禁止轉載。
主站蜘蛛池模板: 亚洲欧美日韩中文无线码| 国产一级二级三级在线观看| 99精品热这里只有精品| 成人小视频免费在线观看| 久久午夜国产片| 最近中文字幕高清中文字幕无| 亚洲欧美人成网站在线观看看| 男人把女人桶到爽爆的视频网站| 午夜视频免费成人| 色爱无码av综合区| 国产免费变态视频网址网站| 欧美黄色一级在线| 国产精品四虎在线观看免费| 91资源在线播放| 在线观看成人免费| h肉3d动漫在线观看网站| 怡红院怡春院首页| 中文在线字幕中文字幕| 日日碰狠狠添天天爽不卡| 久久国产欧美日韩精品| 日韩电影免费在线观看| 亚洲av无码一区二区三区不卡| 欧美成a人片在线观看久| 亚洲欧美日韩久久精品第一区| 激情国产白嫩美女在线观看| 人妻中文字幕无码专区| 精品国产亚洲一区二区三区 | 校花哭着扒开屁股浣肠于柔| 亚洲国产精彩中文乱码av| 欧美狂摸吃奶呻吟| 亚洲欧美中文日韩在线| 波多野结衣丝袜美腿| 亚洲综合色视频在线观看| 特黄特色大片免费播放器999| 天天操天天射天天色| 一区二区视频在线播放| 成人区人妻精品一区二区不卡网站| 中文字幕加勒比| 成人在线综合网| 中国成人在线视频| 性欧美激情xxxd|