本申請屬于人工智能,尤其涉及一種火災預警模型的訓練方法及火災預警系統。
背景技術:
1、機房作為用戶提供通信服務的重要支撐,其失火將會對地區用戶造成重大影響,因此,機房的火災預警系統能否及時準確地發現火災格外重要。
2、為提高機房火災預警的及時性,在相關技術中,通常采用預先訓練好的神經網絡模型來對攝像頭所采集到的機房圖像進行圖像識別,以對機房進行火災預警。
3、然而,相關技術中的神經網絡模型檢測小目標的能力不足,從而在機房中的火焰和/或煙霧較小時,無法及時準確地識別出火災。
技術實現思路
1、本申請實施例提供一種火災預警模型的訓練方法及火災預警系統,能夠及時準確地識別出小目標,保證了機房火災預警的及時性和準確性。
2、第一方面,本申請實施例提供一種火災預警模型的訓練方法,該方法包括:通過火災預警模型的卷積模塊對樣本圖像進行特征提取,得到目標特征向量圖,其中,樣本圖像用于訓練火災預警模型,樣本圖像中至少包含煙霧和/或火以及檢測框數據,檢測框數據至少包括包圍煙霧和/或火焰的檢測框以及檢測框對應的檢測類別標簽,火災預警模型至少包括卷積模塊、初始特征金字塔模塊、擴展特征金字塔模塊以及檢測模塊;通過初始特征金字塔模塊對目標特征向量圖進行特征拼接,得到多個目標特征圖;通過擴展特征金字塔模塊對多個目標特征圖進行特征融合,得到目標融合特征圖;通過檢測模塊對目標融合特征圖進行預測,得到樣本圖像所對應的預測框數據,其中,預測框數據至少包括對樣本圖像中的煙霧和/或火焰進行預測所得到的預測框以及預測框對應的預測類別標簽;根據檢測框數據以及預測框數據之間的差異數據對火災預警模型的模型參數進行更新,得到目標火災預警模型。
3、第二方面,本申請實施例提供了一種火災預警系統,該系統包括圖像采集單元、火災檢測單元以及報警單元,其中,圖像采集單元用于采集火災監控區域內的目標圖像;火災檢測單元用于使用目標火災預警模型對目標圖像進行識別,以識別出目標圖像中的煙霧和/或火焰,其中,目標火災預警模型是采用如第一方面所述的方法訓練得到的模型;報警單元用于在火災檢測單元識別出目標圖像中存在煙霧和/或火焰的情況下,生成火災報警信息。
4、第三方面,本申請實施例提供了一種電子設備,該電子設備包括:處理器以及存儲有計算機程序指令的存儲器;處理器執行計算機程序指令時實現如第一方面所述的火災預警模型的訓練方法。
5、第四方面,本申請實施例提供了一種計算機可讀存儲介質,該計算機可讀存儲介質上存儲有計算機程序指令,計算機程序指令被處理器執行時實現如第一方面所述的火災預警模型的訓練方法。
6、第五方面,本申請實施例提供了一種計算機程序產品,計算機程序產品中的指令由電子設備的處理器執行時,使得電子設備執行如第一方面所述的火災預警模型的訓練方法。
7、本申請對相關技術中的火災預警模型進行了改進,在火災預警模型的檢測層的末尾擴展了一個新的層級,即在原有檢測層中的特征金字塔模塊之后新增了擴展特征金字塔模塊,該擴展特征金字塔模塊可對原有特征金字塔模塊所輸出的特征圖進行特征融合,豐富了小目標的細節,提高了小目標檢測的精度,解決了原有特征金字塔模塊所存在的小目標檢測能力不足的問題,從而使得本申請中的火災預警模型可以準確識別出機房中的煙火和/或火焰等小目標,保證了機房火災預警的及時性和準確性。
1.一種火災預警模型的訓練方法,其特征在于,包括:
2.根據權利要求1所述的方法,其特征在于,所述卷積模塊至少包括下采樣模塊和第一卷積模塊,其中,通過火災預警模型的卷積模塊對樣本圖像進行特征提取,得到目標特征向量圖,包括:
3.根據權利要求2所述的方法,其特征在于,所述初始特征金字塔模塊至少包括第一上采樣模塊、第二卷積模塊以及第一特征拼接模塊,其中,通過所述初始特征金字塔模塊對所述目標特征向量圖進行特征拼接,得到多個目標特征圖,包括:
4.根據權利要求3所述的方法,其特征在于,所述擴展特征金字塔模塊至少包括第二上采樣模塊、特征融合模塊以及第二特征拼接模塊,其中,通過所述擴展特征金字塔模塊對所述多個目標特征圖進行特征融合,得到目標融合特征圖,包括:
5.根據權利要求4所述的方法,其特征在于,通過所述特征融合模塊對所述第一目標特征圖以及所述第二目標特征圖進行特征融合,得到初始融合特征圖,包括:
6.根據權利要求1所述的方法,其特征在于,通過所述檢測模塊對所述目標融合特征圖進行預測,得到所述樣本圖像所對應的預測框數據,包括:
7.根據權利要求1所述的方法,其特征在于,根據所述檢測框數據以及所述預測框數據之間的差異數據對所述火災預警模型的模型參數進行更新,得到目標火災預警模型,包括:
8.一種火災預警系統,其特征在于,包括:圖像采集單元、火災檢測單元以及報警單元,其中,
9.根據權利要求8所述的系統,其特征在于,所述火災檢測單元還用于在確定所述火災監控區域內發生火災的情況下,根據煙霧和/或火焰在所述目標圖像的位置確定所述火災監控區域內的火災發生位置,并確定滅火設備移動至所述火災發生位置的移動軌跡,以使所述滅火設備按照所述移動軌跡移動至所述火災發生位置進行滅火操作。
10.根據權利要求9所述的系統,其特征在于,所述報警單元還用于向目標對象所持有的終端設備發送所述火災報警信息,其中,所述火災報警信息至少包括所述火災發生位置和/或火災發生時間。