基于YOLOv3-tiny的視頻監控目標檢測算法
2022年電子技術應用第7期
王均成1，2，3，賀 超1，2，3，趙志源1，2，3，鄒建紋1，2，3
1.重慶郵電大學 通信與信息工程學院，重慶400065； 2.先進網絡與智能互聯技術重慶市高校重點實驗室，重慶400065；3.泛在感知與互聯重慶市重點實驗室，重慶400065
摘要： 目標檢測算法在視頻監控領域有著較大的實用價值。針對當前在資源受限的視頻監控系統中實現實時目標檢測較為困難的情況，提出了一種基于YOLOv3-tiny改進的目標檢測算法。該算法在YOLOv3-tiny架構的基礎之上，通過添加特征重用來優化骨干網絡結構，并提出全連接注意力混合模塊來學習到更豐富的空間信息，更適合資源約束條件下的目標檢測。實驗數據表明，該算法相比于YOLOv3-tiny在模型體積降低39.2%，參數量降低39.8%，且在VOC數據集上提高了2.7%的mAP，在提高檢測精度的同時顯著降低了模型資源占用。
中圖分類號： TP391.4
文獻標識碼： A
DOI：10.16157/j.issn.0258-7998.212121
中文引用格式： 王均成，賀超，趙志源，等. 基于YOLOv3-tiny的視頻監控目標檢測算法[J].電子技術應用，2022，48(7)：30-33，39.
英文引用格式： Wang Juncheng，He Chao，Zhao Zhiyuan，et al. Video surveillance object detection method based on YOLOv3-tiny[J]. Application of Electronic Technique，2022，48(7)：30-33，39.
文獻標識碼： A
DOI：10.16157/j.issn.0258-7998.212121
中文引用格式： 王均成，賀超，趙志源，等. 基于YOLOv3-tiny的視頻監控目標檢測算法[J].電子技術應用，2022，48(7)：30-33，39.
英文引用格式： Wang Juncheng，He Chao，Zhao Zhiyuan，et al. Video surveillance object detection method based on YOLOv3-tiny[J]. Application of Electronic Technique，2022，48(7)：30-33，39.
Video surveillance object detection method based on YOLOv3-tiny
Wang Juncheng1，2，3，He Chao1，2，3，Zhao Zhiyuan1，2，3，Zou Jianwen1，2，3
1.School of Communication and Information Engineering，Chongqing University of Posts and Telecommunications， Chongqing 400065，China； 2.Advanced Network and Intelligent Connection Technology Key Laboratory of Chongqing Education Commission of China， Chongqing 400065，China； 3.Chongqing Key Laboratory of Ubiquitous Sensing and Networking，Chongqing 400065，China
Abstract： Object detection methods have great value in the application field of video surveillance. At present, it is difficult to realize real-time object detection in resource constrained video surveillance system. A object detection method based on improved YOLOv3-tiny is proposed. Based on the YOLOv3-tiny architecture, the algorithm optimizes the backbone network by adding feature reuse, and a fully-connected attention mix module is proposed to enable the network to learn more abundant spatial information, which is more suitable for object detection under resource constraints. The experimental data shows that compared with YOLOv3-tiny, the algorithm reduces the model volume by 39.2%, the amount of parameters by 39.8%, and improves the mAP of 2.7% on the VOC data set, which significantly reduces the occupation of model resources while improving the detection accuracy.
Key words : object detection；video surveillance；YOLOv3；feature reuse；attention mechanism
0 引言
近年來，目標檢測算法已經廣泛應用于各個視頻監控場景，包括車輛檢測[1]、行人檢測[2]、農業檢測[3]、人類異常行為檢測[4]等，越來越復雜的目標檢測網絡展示了最先進的目標檢測性能。但在實際應用中，往往需要在視頻監控中一些計算能力及內存有限的設備上進行實時目標檢測。例如，嵌入式平臺視頻監控，其可用計算資源一般僅限于低功耗嵌入式圖形處理單元(Graphic Processing Unit，GPU)。這極大地限制了此類網絡在相關領域的廣泛應用，使得在資源受限設備上實現實時目標檢測非常具有挑戰。
為了實現資源有限設備上目標檢測這一挑戰，人們對研究和設計低復雜度的高效神經網絡體系架構越來越感興趣。而著名的YOLO[5](You Only Look Once，YOLO)則是圍繞效率設計的一階段目標檢測算法，它可以在高端圖形處理器上實現視頻監控目標高效檢測。然而對于許多資源受限監控設備來說，這些網絡架構參數量大且計算復雜度較高，使得在嵌入式等監控設備上運行時推理速度大幅下降。YOLOv3[6]是YOLO系列應用在各領域最普遍的算法，YOLOv3-tiny則是在該算法的基礎上簡化的，雖然精度顯著下降但具有了更少計算成本，這大大增加了在資源受限監控設備上部署目標檢測算法的可行性。
本文詳細內容請下載：http://www.viuna.cn/resource/share/2000004582。
作者信息：
王均成1，2，3，賀 超1，2，3，趙志源1，2，3，鄒建紋1，2，3
(1.重慶郵電大學 通信與信息工程學院，重慶400065；
2.先進網絡與智能互聯技術重慶市高校重點實驗室，重慶400065；3.泛在感知與互聯重慶市重點實驗室，重慶400065)
此內容為AET網站原創，未經授權禁止轉載。