[發明專利]隧道擁堵檢測模型的訓練方法、裝置、設備及存儲介質

在審

• 申請號: 202110604652.6
• 申請日: 2021-05-31
• 公開（公告）號: CN115482512A
• 公開（公告）日: 2022-12-16
• 發明（設計）人: 高東生
• 申請（專利權）人: 中國移動通信集團遼寧有限公司;中國移動通信集團有限公司
• 主分類號: G06V20/56
• 分類號: G06V20/56;G06V10/764;G06V10/774;G06V10/82;G06N3/04;G06N3/08
• 代理公司: 北京東方億思知識產權代理有限責任公司 11258
• 代理人: 彭瓊

1.一種隧道擁堵檢測模型的訓練方法，其特征在于，包括：

獲取多個訓練樣本，每個所述訓練樣本包括隧道中的車輛視覺特征信息，以及所述車輛視覺特征信息對應的擁堵類型標簽、以及車燈數量標簽；所述車輛視覺特征信息包括：車輛屬性信息，以及車燈打光效果信息；

將所述車輛屬性信息輸入預設分類模型進行擁堵類型判定，輸出擁堵類型判定結果；以及，將所述車燈打光效果信息輸入預設識別模型進行車燈數量判定，輸出車燈數量判定結果；

統計所述擁堵類型判定結果與所述擁堵類型標簽一致、且所述車燈數量判定結果與車燈數量標簽一致的命中率；

當所述命中率滿足預設閾值時，將所述預設分類模型確定為所述隧道擁堵檢測模型。

2.如權利要求1所述的隧道擁堵檢測模型的訓練方法，其特征在于，所述獲取多個訓練樣本，包括：

獲取隧道圖像；

對所述隧道圖像進行圖像增強處理、歸一化處理，得到所述多個訓練樣本。

3.一種應用隧道擁堵檢測模型檢測隧道擁堵的方法，所述隧道擁堵檢測模型通過權利要求1所述的方法而訓練，所述方法包括：

獲取目標隧道圖像；

將所述目標隧道圖像輸入所述隧道擁堵檢測模型，輸出隧道擁堵檢測結果。

4.如權利要求3所述的應用隧道擁堵檢測模型檢測隧道擁堵的方法，其特征在于，在將所述目標隧道圖像輸入所述隧道擁堵檢測模型之前，所述方法還包括：

對所述目標隧道圖像進行圖像增強處理、歸一化處理。

5.一種隧道擁堵檢測模型的訓練裝置，其特征在于，包括：

獲取模塊，用于獲取多個訓練樣本，每個所述訓練樣本包括隧道中的車輛視覺特征信息，以及所述車輛視覺特征信息對應的擁堵類型標簽、以及車燈數量標簽；所述車輛視覺特征信息包括：車輛屬性信息，以及車燈打光效果信息；

輸出模塊，用于將所述車輛屬性信息輸入預設分類模型進行擁堵類型判定，輸出擁堵類型判定結果；以及，將所述車燈打光效果信息輸入預設識別模型進行車燈數量判定，輸出車燈數量判定結果；

統計模塊，用于統計所述擁堵類型判定結果與所述擁堵類型標簽一致、且所述車燈數量判定結果與車燈數量標簽一致的命中率；

確定模塊，用于當所述命中率不小于所述預設閾值時，將所述預設分類模型確定為所述隧道擁堵檢測模型。

6.如權利要求1所述的隧道擁堵檢測模型的訓練裝置，其特征在于，所述獲取模塊，具體用于：

獲取隧道圖像；

對所述隧道圖像進行圖像增強處理、歸一化處理，得到所述多個訓練樣本。

7.一種應用隧道擁堵檢測模型檢測隧道擁堵的裝置，所述隧道擁堵檢測模型通過權利要求1所述的裝置而訓練，所述裝置包括：

獲取模塊，用于獲取目標隧道圖像；

輸出模塊，用于將所述目標隧道圖像輸入所述隧道擁堵檢測模型，輸出隧道擁堵檢測結果。

8.如權利要求3所述的應用隧道擁堵檢測模型檢測隧道擁堵的裝置，其特征在于，所述裝置還包括：

處理模塊，用于在將所述目標隧道圖像輸入所述隧道擁堵檢測模型之前，對所述目標隧道圖像進行圖像增強處理、歸一化處理。