鄧天民，譚思奇，蒲龍忠

（重慶交通大學交通運輸學院，重慶 400074）

0 概述

近年來，無人駕駛技術(shù)迎來了快速的發(fā)展，對無人駕駛場景中行人［1］、交通標志［2］等目標進行檢測與識別的研究逐漸增多。交通信號燈在無人駕駛場景中扮演著極其重要的角色，因此，交通信號燈的檢測與識別研究具有重要意義。目前基于深度學習的相關(guān)研究較少，主要運用兩階段檢測和一階段檢測這兩類方法。在兩階段檢測中，具有代表性的算法有RCNN［3］、Fast-RCNN［4］、Faster R-CNN［5］等。文獻［6］設(shè)計了一種基于CIFAR-10 的多任務(wù)卷積神經(jīng)網(wǎng)絡(luò)來對復雜環(huán)境下的交通信號燈進行檢測，但是網(wǎng)絡(luò)模型的泛化性和魯棒性不足。兩階段算法需要進行多次的檢測和分類流程，檢測速度相對較慢，因而研究者提出了一階段檢測方法，其中具有代表性的算法有YOLO［7-9］、SSD［10-11］、SqueezeDet［12］等。一階段檢測方法的特點是候選框的產(chǎn)生和分類同時進行，一步到位，檢測速度相對較快。文獻［13］提出了Split-CS-Yolo 算法對交通信號燈進行快速檢測與識別，該方法魯棒性較好但是對黃色信號燈和數(shù)字信號燈的檢測精度還有待提高。文獻［14］對YOLOv3 進行改進，通過在Darknet53 的第2 個殘差塊中增加2 個殘差單元來提高網(wǎng)絡(luò)對小目標的檢測性能，所提出的改進網(wǎng)絡(luò)能夠達到較高的檢測精度，但不滿足實時性的要求。可見，以上改進方法對交通信號燈的檢測效果都不能滿足實際需求，在檢測速度和檢測精度之間難以達到較好的平衡。

本文通過改進YOLOv5s 算法，提升對交通信號燈的識別精度和檢測速度。設(shè)計一種基于密集連接的CSP 殘差模塊來替換原網(wǎng)絡(luò)模型主干網(wǎng)絡(luò)中的殘差結(jié)構(gòu)，使得主干網(wǎng)絡(luò)能更準確地對信號燈的特征進行提取，進而提高網(wǎng)絡(luò)的檢測精度。同時，設(shè)計一種基于多層次的跨連接CSP 殘差模塊來替換頸部網(wǎng)絡(luò)中的殘差結(jié)構(gòu)，增強網(wǎng)絡(luò)的特征融合能力。為加快網(wǎng)絡(luò)模型的檢測速度，對檢測尺度進行改進，考慮交通信號燈屬于小目標的屬性，只保留小目標檢測尺度。在此基礎(chǔ)上，采用巴黎交通信號燈LaRA 數(shù)據(jù)集進行實驗，驗證本文方法的有效性。

1 YOLOv5s 網(wǎng)絡(luò)模型

YOLOv5 網(wǎng)絡(luò)共有4 個版本，分別為YOLOv5s、YOLOv5m、YOLOv5l、YOLOv5x。YOLOv5s 是該系列檢測網(wǎng)絡(luò)中深度和特征圖寬度最小的網(wǎng)絡(luò)，其他幾個版本的網(wǎng)絡(luò)都是在此基礎(chǔ)上進行加深和加寬的。相比于其他系列的YOLO 檢測網(wǎng)絡(luò)，YOLOv5的權(quán)重文件相對較小，權(quán)重最小的YOLOv5s 僅27 MB，權(quán)重最大的YOLOv5x 也只有170 MB。因此，基于體積小的優(yōu)勢，YOLOv5 非常適合部署到嵌入式的設(shè)備中。實現(xiàn)對交通信號燈的檢測是無人駕駛的重要環(huán)節(jié)，而YOLOv5 非常適合應(yīng)用在交通場景下對交通信號燈進行檢測［15］?；趯?quán)重文件大小、識別精度和檢測速度的綜合考慮，本文選擇檢測速度較快、識別精度較高的YOLOv5s 進行研究，對其網(wǎng)絡(luò)結(jié)構(gòu)進行改進。

YOLOv5s 網(wǎng)絡(luò)結(jié)構(gòu)如圖1 所示，其中的主干網(wǎng)絡(luò)對輸入的圖像進行了5 次下采樣來提取圖像的特征，主干網(wǎng)絡(luò)包含F(xiàn)ocus、CBH、CSP1-x、SPP［16］4 種模塊。

圖1 YOLOv5s 網(wǎng)絡(luò)結(jié)構(gòu)Fig.1 YOLOv5s network structure

輸入YOLOv5s 網(wǎng)絡(luò)的圖像通過Focus 模塊進行自我復制，然后進行切片操作，以此來減少網(wǎng)絡(luò)的計算量，加快對候選區(qū)域特征提取的速度。切片的圖像通過CBH 模塊進行卷積、歸一化和激活操作，然后進行特征提取。在特征提取的過程中，CSP1-x 殘差結(jié)構(gòu)用于優(yōu)化網(wǎng)絡(luò)中的梯度信息，減少推理計算量，加快計算速度。最后，SPP 模塊將不同尺寸的輸入轉(zhuǎn)換為相同大小的輸出，解決輸入圖像尺寸不統(tǒng)一的問題。YOLOv5 的Neck 網(wǎng)絡(luò)使用了ReLU 激活函數(shù)，采用FPN［17］+PAN［18］網(wǎng)絡(luò)結(jié)構(gòu)，其中FPN 層通過自頂向下對圖像進行上采樣，將提取到的特征與主干網(wǎng)絡(luò)中提取到的特征進行融合。在FPN 層之后，還添加了一個自底向上的特征金字塔結(jié)構(gòu)，用于對圖像進行降采樣，將提取到的特征和FPN 層提取到的特征進行融合。通過FPN+PAN 網(wǎng)絡(luò)結(jié)構(gòu)可以對主干網(wǎng)絡(luò)和檢測網(wǎng)絡(luò)提取的特征進行聚合，增強網(wǎng)絡(luò)特征融合的能力。此外，Neck 網(wǎng)絡(luò)中還加入了CSP2-x 結(jié)構(gòu)對提取的特征進行融合。輸出端采用CIoU［19］作為Bounding box 的損失函數(shù)，并且提供了3 種不同的檢測尺度（20×20、40×40、80×80）。

2 YOLOv5s 網(wǎng)絡(luò)模型改進

2.1 CSPD-x 殘差結(jié)構(gòu)改進

YOLOv5s 網(wǎng)絡(luò)中有3處CSP1-x 結(jié)構(gòu)，該結(jié)構(gòu)借鑒了CSPNet［20］的思路，把基礎(chǔ)層的特征映射為2個部分，然后通過跨階段層次結(jié)構(gòu)進行合并，從而減少計算量和提高運行速度，同時保證精度。CSP1-x 殘差結(jié)構(gòu)如圖2 所示。

圖2 CSP1-x 殘差結(jié)構(gòu)Fig.2 CSP1-x residual structure

CSP1-x殘差組件的設(shè)計借鑒了ResNet［21］網(wǎng)絡(luò)的設(shè)計思想，通過在卷積層之間增加跨連接（Shortcut）方式，減少網(wǎng)絡(luò)模型的計算量，加快運行效率?？邕B接（Shortcut）和CSP1-x殘差組件結(jié)構(gòu)分別如圖3、圖4所示。

圖3 Shortcut 連接Fig.3 Shortcut connection

圖4 CSP1-x 殘差組件結(jié)構(gòu)Fig.4 CSP1-x residual component structure

為了更好地解決梯度發(fā)散的問題，文獻［22］提出了DenseNet 網(wǎng)絡(luò)。在DenseNet 中，每兩個卷積層之間都添加了跨連接，密集跨連接的方式使得各卷積層獲取的特征都能向后進行傳遞并且相互融合，因此，提取到的特征也更豐富和多樣化。DenseNet核心表達式如式（1）所示：

其中：xk表示第k層網(wǎng)絡(luò)的輸出；[x0,x1,…,xk-1]表示將第0 層到第k-1 層網(wǎng)絡(luò)提取的特征進行合并；Hk表示包括卷積層、歸一化層和ReLu 激活函數(shù)層的組合方式。DenseNet 結(jié)構(gòu)如圖5 所示。

圖5 DenseNet 網(wǎng)絡(luò)結(jié)構(gòu)Fig.5 DenseNet network structure

基于DenseNet 網(wǎng)絡(luò)結(jié)構(gòu)，本文設(shè)計了新的unit組件D-unit（Dense-unit）。D-unit依舊采用1×1 和3×3兩種大小的卷積核、歸一化層和Hardwish 激活函數(shù)層。通過在原來的基礎(chǔ)上對每個CBH 模塊之間都添加跨連接，使得各個模塊緊密連接，從而更好地解決深度網(wǎng)絡(luò)模型退化問題。D-unit 殘差組件結(jié)構(gòu)如圖6 所示。

圖6 D-unit 殘差組件結(jié)構(gòu)Fig.6 D-unit residual component structure

首先，輸入的特征信息經(jīng)過第1 個CBH 模塊，該模塊含有1×1 大小的卷積核；然后，將卷積后得到的特征信息與輸入的信息進行融合輸出到第2 個CBH模塊進行卷積，第2 個CBH 模塊采用3×3 大小的卷積核，能得到與第1 個CBH 模塊不同的特征信息；最后，將最開始輸入到D-unit 的特征信息與第1 個CBH 模塊和第2 個CBH 模塊得到的特征信息進行融合，將融合后的特征信息輸入到下一個特征提取結(jié)構(gòu)中。

改進后CSPD-x 的殘差組件以跨連接的方式將各卷積層進行密集連接，主要是為了防止模型的退化，使得模型可以更快地收斂到最優(yōu)解。密集連接能夠?qū)μ卣鬟M行重利用，使特征相互融合，而對融合后的特征進行提取能夠獲得更多的目標信息，提高識別的精度。CSPD-x 結(jié)構(gòu)如圖7 所示。

圖7 CSPD-x 殘差結(jié)構(gòu)Fig.7 CSPD-x residual structure

2.2 CSPS-x 殘差結(jié)構(gòu)改進

YOLOv5s 網(wǎng)絡(luò)中還使用了另一種殘差結(jié)構(gòu)——CSP2-x。與CSP1-x 結(jié)構(gòu)不同，CSP2-x 結(jié)構(gòu)主要是對輸入的特征圖進行卷積，并對所提取到的特征信息進行融合。在CSP2-x 結(jié)構(gòu)中，殘差組件的作用僅僅是對輸入圖像的特征信息進行提取。CSP2-x 殘差結(jié)構(gòu)如圖8 所示。

圖8 CSP2-x 殘差結(jié)構(gòu)Fig.8 CSP2-x residual structure

通過對原殘差組件的分析，本文設(shè)計了一種新的殘差組件S-unit（Shortcut-unit）。該結(jié)構(gòu)由4 個CBH 模塊組成，卷積核的大小分別對應(yīng)為1×1、3×3、1×1、3×3。各個模塊之間通過多個跨連接方式相互連接。分別在第1 個和第2 個、第3 個和第4 個CBH模塊間加入跨連接，然后再將第2 個和第4 個CBH模塊之間進行跨連接。原始輸入的特征信息通過第1 個卷積核大小為1×1 的CBH 模塊，提取后的特征信息輸入到第2 個卷積核大小為3×3 的CBH 模塊，將原始輸入的特征信息和第2 個CBH 模塊所提取到的信息進行融合再輸入到第3 個卷積核大小為1×1 的CBH 模塊，得到的特征信息和第1 個CBH 模塊所提取到的信息融合后輸入到第4 個卷積核大小為3×3 的CBH 模塊，第4 個CBH 模塊輸出的結(jié)果和第1 個CBH 模塊所得到的特征信息融合后再輸入到后面的卷積網(wǎng)絡(luò)層。S-unit 殘差組件與原網(wǎng)絡(luò)殘差組件對比如圖9 所示。

圖9 S-unit 殘差組件和原網(wǎng)絡(luò)殘差組件結(jié)構(gòu)Fig.9 S-unit residual component structure and the original network residual component structure

在原結(jié)構(gòu)中，CSP2-x 殘差組件的作用主要是對特征圖的特征信息進行提取，各卷積層之間的特征信息并沒有相互融合，這樣會導致隨著卷積層層數(shù)增加提取到的特征信息部分丟失。此外，原結(jié)構(gòu)直接將各卷積層連接起來，會導致訓練時網(wǎng)絡(luò)模型的退化，且梯度容易發(fā)散，難以收斂到最優(yōu)的結(jié)果。CSPS-x 殘差結(jié)構(gòu)通過跨連接的方式對各層提取到的特征信息進行融合，有效地解決了特征信息丟失的問題，而且還能防止模型的退化，使得對網(wǎng)絡(luò)的訓練更容易進行。CSPS-x 殘差結(jié)構(gòu)如圖10 所示。

圖10 CSPS-x 殘差結(jié)構(gòu)Fig.10 CSPS-x residual structure

2.3 檢測尺度改進

原YOLOv5s 網(wǎng)絡(luò)共有3 種規(guī)格的檢測尺度，分別為20×20、40×40、80×80。20×20 檢測尺度的感受野為32×32 大小的像素區(qū)域，主要用于檢測大型物體；40×40檢測尺度的感受野為16×16 大小的像素區(qū)域，主要用于檢測中等大小的物體；80×80檢測尺度的感受野為8×8大小的像素區(qū)域，主要用于檢測小型物體。

由于交通信號燈在交通場景下屬于較小的目標，因此本文對原YOLOv5s 的3 種檢測尺度進行改進，保留40×40 和80×80 這兩種針對中小型目標的檢測尺度，去除20×20 這一針對大型目標的檢測尺度，從而使網(wǎng)絡(luò)能夠更好地處理待檢測目標，同時加快網(wǎng)絡(luò)的運行速度。改進后的網(wǎng)絡(luò)命名為TL-YOLOv5s，結(jié)構(gòu)如圖11所示。

圖11 TL-YOLOv5s 網(wǎng)絡(luò)結(jié)構(gòu)Fig.11 TL-YOLOv5s network structure

3 實驗

法國巴黎LaRA 信號燈數(shù)據(jù)集適用于目標檢測研究，且只標注了交通信號燈這一種類別，數(shù)據(jù)量大，包含的交通場景也非常豐富，因此，本文選用LaRA 信號燈數(shù)據(jù)集對所提出的方法進行評估。對于實驗部分使用的硬件平臺，CPU 采用Intel i7-9750H，GPU 采用NVIDAI GTX1660Ti，操作系統(tǒng)為Window10，訓練框架采用Pytorch1.6。

3.1 實驗數(shù)據(jù)集

法國巴黎LaRA 信號燈數(shù)據(jù)集包含9 168 張圖片，圖片尺寸為640×480像素，包含green（通行）、orange（警示）、red（停止）、ambiguous（模糊）4 種不同類別的標簽，數(shù)量分別為3 381、58、5 280、449 張。其中，模糊的圖片是由于在拍攝的過程中車輛出現(xiàn)抖動所導致。各信號燈種類及標簽數(shù)量如表1 所示。

表1 LaRA 信號燈數(shù)據(jù)集中各類目標的數(shù)量Table 1 The number of various targets in LaRA signal light dataset

由于數(shù)據(jù)集中包含的模糊變形的圖片會對實驗的結(jié)果產(chǎn)生較大影響，因此本文對LaRA數(shù)據(jù)集進行處理，剔除掉449 張模糊變形的圖片，使用剩余的8 719 張圖片進行實驗。數(shù)據(jù)集圖片樣本如圖12 所示。

圖12 LaRA 數(shù)據(jù)集圖片樣本Fig.12 Image samples of LaRA dataset

3.2 模型訓練

對原YOLOv5s 模型和改進后的YOLOv5s 模型分別進行訓練。剔除LaRA 信號燈數(shù)據(jù)集中模糊的449 張圖片和標簽，將剩余的圖片和標簽按8∶1∶1 的比例分為訓練集、測試集和驗證集，訓練階段初始學習率調(diào)整為0.01，訓練迭代次數(shù)epoch 設(shè)定為200 次。在此基礎(chǔ)上，通過GPU 對原YOLOv5s 和改進后的YOLOv5s 網(wǎng)絡(luò)模型進行訓練。

3.3 評價指標

本文選取精確率（Precision）、召回率（Recall）、平均精度（Average Precision，AP）、平均精度均值（mean Average Precision，mAP）、幀率（Frames Per Second，F(xiàn)PS）等評價指標［23］對模型性能進行評價。

1）準確率Pprecision和召回率Rrecall的計算公式如下：

其中：TP表示正確檢測出的目標數(shù)量；FP表示被誤檢的目標數(shù)量；FN表示未被檢測出的樣本數(shù)量。

2）平均精度AAP和平均精度均值mmAP的計算公式如下：

其中：Nc表示類別個數(shù)；PA表示不同類別的平均精度。實驗結(jié)束后利用實驗數(shù)據(jù)可以繪制出網(wǎng)絡(luò)模型的PR 曲線，該曲線所圍成的面積即被定義為AP，用于評估模型在單個檢測類別上的表現(xiàn)。計算出每一類的AP 值之后，進行平均即得到mAP。通常來說，mAP 值越高，表示網(wǎng)絡(luò)模型性能越好。

3）檢測速度一般用FPS 來衡量，表示目標檢測網(wǎng)絡(luò)每秒能處理圖片的數(shù)量，F(xiàn)PS 值越大，表示網(wǎng)絡(luò)模型處理圖像的速度越快。

3.4 實驗結(jié)果及分析

3.4.1 主干網(wǎng)絡(luò)殘差結(jié)構(gòu)改進實驗

原YOLOv5s 主干網(wǎng)絡(luò)中共有3 處CSP1-x 結(jié)構(gòu)，對其進行替換加入新的結(jié)構(gòu)CSPD-x，然后再對改進后的網(wǎng)絡(luò)模型和原網(wǎng)絡(luò)模型分別進行訓練，在LaRA數(shù)據(jù)集上比較2 種網(wǎng)絡(luò)的檢測結(jié)果。對改進后模型訓練200 個epoch 后損失曲線變化如圖13 所示，其中cls_loss、obj_loss 和box_loss 分別代表分類損失、置信度損失和邊界框回歸損失，由該圖可以看出曲線走勢基本趨于平穩(wěn)。

圖13 模型改進后的訓練損失曲線Fig.13 Training loss curve after model improvement

原網(wǎng)絡(luò)和改進網(wǎng)絡(luò)（以CSPD-x 表示）識別各類別的平均精度和平均精度均值如表2 所示。

表2 主干網(wǎng)絡(luò)殘差結(jié)構(gòu)改進前后實驗結(jié)果對比Table 2 Comparison of experimental results before and after improvement of residual structure in backbone network %

由表2 可以看出：改進后的網(wǎng)絡(luò)相比于原網(wǎng)絡(luò)對各類目標的識別精度都有提升，其中綠色信號燈的識別精度相比于原網(wǎng)絡(luò)提升了7.2 個百分點，由于數(shù)據(jù)集中黃色信號燈數(shù)量較少且圖片較為清楚，因此黃色信號燈的識別精度較高，而紅色信號燈的識別精度提升不是特別明顯，僅提升了0.3 個百分點，但是總體上改進網(wǎng)絡(luò)mAP 值較原網(wǎng)絡(luò)提升了2.5 個百分點。

3.4.2 特征融合網(wǎng)絡(luò)殘差結(jié)構(gòu)改進實驗

為驗證不同跨連接方式的效果，設(shè)計另外2 種不同的殘差組件來進行對比實驗（以CSPS1-x 和CSPS2-x 表示），驗證CSPS-x 結(jié)構(gòu)的有效性，各殘差組件結(jié)構(gòu)如圖14 所示。最后通過比較和分析選擇出最優(yōu)的連接方案對原CSP2-x 結(jié)構(gòu)進行替換。

圖14 3 種殘差組件結(jié)構(gòu)對比Fig.14 Comparison of three residual component structures

根據(jù)3 種不同的改進方案，分別對原CSP2-x 結(jié)構(gòu)進行替換和訓練，訓練完成后進行測試，實驗結(jié)果如表3 所示。

表3 特征融合網(wǎng)絡(luò)殘差結(jié)構(gòu)改進前后實驗結(jié)果對比Table 3 Comparison of experimental results before and after improvement of residual structure in feature fusion network %

由表3 可以看出：相較于原網(wǎng)絡(luò)，3 種改進網(wǎng)絡(luò)識別效果都有所提升。在對綠色信號燈的識別上，CSPS-x 性能最好（AP 為66.8%），相比于原網(wǎng)絡(luò)提升10.8 個百分點；在對黃色信號燈的識別上，由于各網(wǎng)絡(luò)的識別效果都較好（AP 均為99.5%），因此3 種改進網(wǎng)絡(luò)的AP 值沒有提升；在對紅色信號燈的識別上，各網(wǎng)絡(luò)的效果都較差，3 種改進網(wǎng)絡(luò)的AP 只有小幅提升，CSPR1-x 相比于原網(wǎng)絡(luò)提升0.1 個百分點（AP 為36%），CSPR2-x 和CSPS-x 都提升0.2 個百分點（AP 為36.1%）。總體來看，3 種改進方案對交通信號燈識別的mAP 值都有較大的提升，其中第3 種方案提升最大（mAP 為67.5%），相較于原網(wǎng)絡(luò)提升3.7 個百分點。由此可見，CSPS-x 結(jié)構(gòu)所帶來的優(yōu)化效果優(yōu)于CSPS1-x 和CSPS2-x。因此，選擇CSPS-x結(jié)構(gòu)對原網(wǎng)絡(luò)進行改進，替換原網(wǎng)絡(luò)中的CSP2-x結(jié)構(gòu)。

3.4.3 檢測尺度改進實驗

本文對原網(wǎng)絡(luò)進行了檢測尺度的改進。原YOLOv5s 共有3 種檢測尺度，分別為20×20、40×40、80×80。本實驗通過增加或減少檢測尺度的種類來尋找出最適合交通信號燈檢測的尺度種類。

為驗證本文對于檢測尺度改進的合理性和有效性，在原網(wǎng)絡(luò)的基礎(chǔ)上去除20×20、40×40 規(guī)格的檢測尺度，保留80×80 的檢測尺度，得到只具有一個檢測尺度的網(wǎng)絡(luò)來增加實驗的可對比性。然后再根據(jù)之前的改進思路在原網(wǎng)絡(luò)的基礎(chǔ)上去掉20×20 的檢測尺度，保留40×40、80×80 這2 個規(guī)格檢測尺度。最后對原網(wǎng)絡(luò)和檢測尺度改進后的網(wǎng)絡(luò)模型分別進行訓練，得到的檢測結(jié)果如表4 所示

表4 檢測尺度改進前后實驗結(jié)果對比Table 4 Comparison of experimental results before and after improvement of detection scale

由表4 可以看出：只有1 個80×80 檢測尺度的網(wǎng)絡(luò)速度最快（36.5 frame/s），相比于原網(wǎng)絡(luò)提升27.6 個百分點，但mAP 值下降1.4 個百分點；具有2 個檢測尺度的改進網(wǎng)絡(luò)檢測速度為32.3 frame/s，相比于原網(wǎng)絡(luò)提升12.9 個百分點，而mAP 值只減少0.3 個百分點；1 個檢測尺度的網(wǎng)絡(luò)相比于2 個檢測尺度的網(wǎng)絡(luò)識別精度減少較大，同時相比于原網(wǎng)絡(luò)，改進后的2 個網(wǎng)絡(luò)都能達到較高的檢測速率。綜合以上實驗結(jié)果，本文選擇保留40×40、80×80 這2 種規(guī)格的檢測尺度來對原網(wǎng)絡(luò)進行改進。

3.4.4 網(wǎng)絡(luò)模型消融實驗

通過以上實驗分析，對原網(wǎng)絡(luò)的網(wǎng)絡(luò)結(jié)構(gòu)分步驟依次進行改進，對改進后的網(wǎng)絡(luò)模型進行訓練并測試，消融實驗結(jié)果如表5 所示。其中：●表示應(yīng)用此方案；○表示不應(yīng)用此方案。

表5 消融實驗結(jié)果對比Table 5 Comparison of results in ablation experiment

由表5可以看出：在采用CSPD-x結(jié)構(gòu)和CSPS-x結(jié)構(gòu)分別替換了CSP1-x 結(jié)構(gòu)和CSP2-x 結(jié)構(gòu)后，各類別識別精度都有較大的提升，mAP 值達到70.5%，但是檢測速度為19.9 frame/s，相比于原網(wǎng)絡(luò)下降8.7 frame/s；在去除20×20 檢測尺度后，各類別的識別精度略微下降，mAP 值降低了0.4 個百分點，但檢測速度較未去除之前提升12.6 個百分點，達到22.4 frame/s。由此可見，改進網(wǎng)絡(luò)能夠較好地實現(xiàn)對交通信號燈的檢測與識別，并且滿足實時性的要求。

實際環(huán)境下的檢測效果對比如圖15 所示，從中可以看出，原網(wǎng)絡(luò)存在許多信號燈未被檢測出的情況，而改進網(wǎng)絡(luò)對交通信號燈的檢測效果更好，識別精度更高。由此可見，改進網(wǎng)絡(luò)更適用于對交通信號燈的檢測與識別。

在LaRA 數(shù)據(jù)集上將本文方法與使用不同網(wǎng)絡(luò)的識別方法進行對比，如表6 所示。

表6 LaRA 數(shù)據(jù)集上不同方法的實驗結(jié)果對比Table 6 Comparison of experimental results by different methods in LaRA dataset %

由 表6 可以看出：TL-YOLOv5s 網(wǎng)絡(luò)在LaRA 數(shù)據(jù)集上相比于其他網(wǎng)絡(luò)具有更好的識別性能，對黃色和綠色信號燈的識別精度均為最高，對紅色信號燈的識別精度略低于Faster R-CNN?？傮w上，TL-YOLOv5s 網(wǎng)絡(luò)模型的mAP 值達到70.1%，相比于RefineDet、Faster R-CNN、R-FCN 等方法有較大幅度提升。

4 結(jié)束語

針對傳統(tǒng)交通信號燈識別方法精度低且難以達到實時檢測要求的問題，本文提出一種改進的YOLOv5s 網(wǎng)絡(luò)。設(shè)計2 種新的殘差模塊替換YOLOv5s 網(wǎng)絡(luò)中的殘差模塊，增強主干網(wǎng)絡(luò)特征提取能力和融合網(wǎng)絡(luò)特征融合能力，提高檢測精度同時防止網(wǎng)絡(luò)訓練時模型退化。在此基礎(chǔ)上，對網(wǎng)絡(luò)的檢測尺度進行改進，去除大目標檢測尺度而保留中小目標檢測尺度，以此來適應(yīng)交通信號燈的檢測，進一步提高網(wǎng)絡(luò)運行的速度。實驗結(jié)果表明，TL-YOLOv5s 在LaRA 數(shù)據(jù)集上取得了70.10%的平均精度均值和22.38 frame/s 的檢測速度，相比YOLOv5s、R-FCN 等網(wǎng)絡(luò)具有更高的識別精度且滿足實時性的要求。下一步將通過增加數(shù)據(jù)集中交通信號燈的類別擴大模型的識別范圍，提高其在實際環(huán)境下的檢測性能。