本發(fā)明屬于圖像數(shù)據(jù)處理,具體涉及一種基于人工智能的高空墜物追溯系統(tǒng)。
背景技術(shù):
1、高空墜物事故在城市高層建筑中時有發(fā)生,嚴重威脅到行人的生命財產(chǎn)安全。隨著城市化進程的加快,高層建筑越來越多,因高空拋物或墜物引發(fā)的傷害事件也逐年增多。這類事故不僅會造成直接的人員傷亡和財產(chǎn)損失,還可能引發(fā)法律糾紛和社會矛盾,已成為城市安全管理中的一大難題。
2、目前,針對高空墜物的追溯方法主要依賴于目擊者證言和現(xiàn)場監(jiān)控錄像。這些方法在一定程度上能夠幫助確定墜物的來源,但普遍存在效率低、準確性差、取證困難等問題。目擊者證言往往因視角局限或記憶模糊而不夠可靠,根據(jù)監(jiān)控錄像追溯則耗時費力,且容易因視頻質(zhì)量低下或信息缺失導(dǎo)致結(jié)果不準確。
3、因此,亟須一種高空墜物追溯系統(tǒng)以解決上述問題。
技術(shù)實現(xiàn)思路
1、本發(fā)明提供一種基于人工智能的高空墜物追溯系統(tǒng),解決相關(guān)技術(shù)中效率低、準確性差、取證困難以及人力成本高的技術(shù)問題。
2、本發(fā)明提供了一種基于人工智能的高空墜物追溯系統(tǒng),包括:
3、圖像采集模塊,其用于采集目標物的圖像數(shù)據(jù);
4、圖像數(shù)據(jù)包括從目標物出現(xiàn)到目標物消失或落地的視頻數(shù)據(jù);
5、預(yù)處理模塊,其用于對圖像數(shù)據(jù)進行預(yù)處理獲得第一目標圖像,其中第一目標圖像包含n個連續(xù)的視頻幀,n為自定義參數(shù);
6、目標檢測模塊,其用于根據(jù)第一目標圖像構(gòu)建目標檢測模型,目標檢測模型的輸入為第一目標圖像,輸出為第一特征序列,其中第一特征序列的第n個序列單元表示第n個視頻幀的目標物類別、邊界框?qū)挾?、邊界框高度、邊界框中心點位置以及置信度,其中1≤n≤n;
7、軌跡生成模塊,其用于根據(jù)第一特征序列結(jié)合跟蹤算法生成目標物的運動軌跡。
8、進一步地,預(yù)處理的步驟包括:
9、步驟s201,通過對圖像數(shù)據(jù)進行噪聲去除處理,其中σ是高斯濾波器的標準差,α表示高斯濾波器的權(quán)重系數(shù),gx,y表示位于x,y的高斯函數(shù)值,即該點的權(quán)重;
10、步驟s202,通過對圖像數(shù)據(jù)進行增強處理,其中hu,v表示濾波器的頻率響應(yīng),du,v表示頻率域中的距離函數(shù),d0表示截止頻率,n表示濾波器的階數(shù),用于控制濾波器在截止頻率附近的陡峭程度,β表示高通濾波器的權(quán)重系數(shù);
11、步驟s203,通過分離出目標物,并將其在視頻幀中框選出,其中itx,y表示視頻幀圖像,x,y為空間坐標,表示圖像中的像素位置,bx,y表示背景模型,itx,y-bx,y表示每一幀圖像與背景模型之間的像素差異,t表示第一閾值,mx,y表示目標物掩碼,目標物掩碼由1和0組成;
12、步驟s204,將圖像數(shù)據(jù)按照時間順序逆序排列;
13、步驟s205,對圖像數(shù)據(jù)通過z分數(shù)歸一化方法進行歸一化處理。
14、進一步地,目標檢測模型基于卷積神經(jīng)網(wǎng)絡(luò)構(gòu)建,通過對目標檢測模型訓(xùn)練得到目標物的類別、邊界框的寬度、邊界框高度、邊界框中心點位置和置信度,邊界框表示包含目標物在內(nèi)的矩形,置信度表示目標物在邊界框內(nèi)的概率,同時設(shè)置置信度閾值,對于低于置信度閾值的視頻幀,不采用其輸出的目標物類別、邊界框?qū)挾?、邊界框高度和邊界框中心點位置,通過線性插值法替換所述視頻幀經(jīng)過目標檢測模型輸出的目標物類別、邊界框?qū)挾取⑦吔缈蚋叨群瓦吔缈蛑行狞c位置。
15、進一步地,對于低于置信度閾值的視頻幀的目標物類別、邊界框?qū)挾取⑦吔缈蚋叨群瓦吔缈蛑行狞c位置通過線性插值法獲取,具體包括:將距離低置信度幀最近的高置信度幀的目標物類別作為低置信度幀的目標物類別;將低置信度幀之前的最近的高置信度幀和之后的最近的高置信度幀作為插值的起點和終點,并據(jù)此計算低置信度幀的邊界框的寬度、邊界框的高度和邊界框中心點的位置。
16、進一步地,通過人工模擬的方式獲取目標檢測模型的樣本數(shù)據(jù),通過人工標注的方式獲取目標檢測模型的樣本標簽。
17、進一步地,在目標檢測模型訓(xùn)練之前先對第一目標圖像的n個視頻幀進行預(yù)訓(xùn)練,在預(yù)訓(xùn)練的過程中,將第一目標圖像的n個視頻幀輸入到分類器中,分類器的分類空間表示目標物的類別。
18、進一步地,目標檢測模型的損失函數(shù)包括:用于分類的交叉熵損失函數(shù)、用于邊界框的l1損失函數(shù)和用于置信度的二元交叉熵損失函數(shù),總損失函數(shù)為:ltotal=λclslcls+λreglreg+λconflconf,其中l(wèi)cls、lreg和lconf分別為交叉熵損失函數(shù)、l1損失函數(shù)和二元交叉熵損失函數(shù),λcls、λreg和λconf分別為交叉熵損失函數(shù)、l1損失函數(shù)和二元交叉熵損失函數(shù)的權(quán)重參數(shù)。
19、進一步地,使用目標檢測模型第一次輸出的結(jié)果初始化跟蹤器,設(shè)置目標物的初始位置及邊界框的寬度和高度,并在隨后的每一幀中通過跟蹤算法預(yù)測目標物的新位置,將其與目標檢測獲得的邊界框中心點位置結(jié)合,更新目標物在視頻幀中的位置,并將每一幀的目標物位置保存到軌跡列表里,從而形成目標物的運動軌跡。
20、本發(fā)明還提供一種計算設(shè)備,包括:
21、一個或多個處理器;
22、存儲裝置,用于存儲一個或多個程序,當(dāng)所述一個或多個程序被所述一個或多個處理器執(zhí)行,使得所述一個或多個處理器實現(xiàn)所述的系統(tǒng)。
23、本發(fā)明還提供一種計算機可讀存儲介質(zhì),所述計算機可讀存儲介質(zhì)中存儲有程序,該程序被處理器執(zhí)行時實現(xiàn)所述的系統(tǒng)。
24、本發(fā)明的有益效果在于:本發(fā)明通過構(gòu)建并訓(xùn)練目標檢測模型得到目標物邊界框中心點在圖像中每一幀的位置,并結(jié)合跟蹤算法更新目標物位置,提高了目標物源頭位置的準確性。
1.一種基于人工智能的高空墜物追溯系統(tǒng),其特征在于,包括:
2.根據(jù)權(quán)利要求1所述的一種基于人工智能的高空墜物追溯系統(tǒng),其特征在于,預(yù)處理的步驟包括:
3.根據(jù)權(quán)利要求1所述的一種基于人工智能的高空墜物追溯系統(tǒng),其特征在于,目標檢測模型基于卷積神經(jīng)網(wǎng)絡(luò)構(gòu)建,通過對目標檢測模型訓(xùn)練得到目標物的類別、邊界框的寬度、邊界框高度、邊界框中心點位置和置信度,邊界框表示包含目標物在內(nèi)的矩形,置信度表示目標物在邊界框內(nèi)的概率,同時設(shè)置置信度閾值,對于低于置信度閾值的視頻幀,不采用其輸出的目標物類別、邊界框?qū)挾?、邊界框高度和邊界框中心點位置,通過線性插值法替換所述視頻幀經(jīng)過目標檢測模型輸出的目標物類別、邊界框?qū)挾?、邊界框高度和邊界框中心點位置。
4.根據(jù)權(quán)利要求3所述的一種基于人工智能的高空墜物追溯系統(tǒng),其特征在于,對于低于置信度閾值的視頻幀的目標物類別、邊界框?qū)挾?、邊界框高度和邊界框中心點位置通過線性插值法獲取,具體包括:將距離低置信度幀最近的高置信度幀的目標物類別作為低置信度幀的目標物類別;將低置信度幀之前的最近的高置信度幀和之后的最近的高置信度幀作為插值的起點和終點,并據(jù)此計算低置信度幀的邊界框的寬度、邊界框的高度和邊界框中心點的位置。
5.根據(jù)權(quán)利要求1所述的一種基于人工智能的高空墜物追溯系統(tǒng),其特征在于,通過人工模擬的方式獲取目標檢測模型的樣本數(shù)據(jù),通過人工標注的方式獲取目標檢測模型的樣本標簽。
6.根據(jù)權(quán)利要求1所述的一種基于人工智能的高空墜物追溯系統(tǒng),其特征在于,在目標檢測模型訓(xùn)練之前先對第一目標圖像的n個視頻幀進行預(yù)訓(xùn)練,在預(yù)訓(xùn)練的過程中,將第一目標圖像的n個視頻幀輸入到分類器中,分類器的分類空間表示目標物的類別。
7.根據(jù)權(quán)利要求1所述的一種基于人工智能的高空墜物追溯系統(tǒng),其特征在于,目標檢測模型的損失函數(shù)包括:用于分類的交叉熵損失函數(shù)、用于邊界框的l1損失函數(shù)和用于置信度的二元交叉熵損失函數(shù),總損失函數(shù)為:ltotal=λclslcls+λreglreg+λconflconf,其中l(wèi)cls、lreg和lconf分別為交叉熵損失函數(shù)、l1損失函數(shù)和二元交叉熵損失函數(shù),λcls、λreg和λconf分別為交叉熵損失函數(shù)、l1損失函數(shù)和二元交叉熵損失函數(shù)的權(quán)重參數(shù)。
8.根據(jù)權(quán)利要求1所述的一種基于人工智能的高空墜物追溯系統(tǒng),其特征在于,使用目標檢測模型第一次輸出的結(jié)果初始化跟蹤器,設(shè)置目標物的初始位置及邊界框的寬度和高度,并在隨后的每一幀中通過跟蹤算法預(yù)測目標物的新位置,將其與目標檢測獲得的邊界框中心點位置結(jié)合,更新目標物在視頻幀中的位置,并將每一幀的目標物位置保存到軌跡列表里,從而形成目標物的運動軌跡。
9.一種計算設(shè)備,其特征在于,包括:
10.一種計算機可讀存儲介質(zhì),其特征在于,所述計算機可讀存儲介質(zhì)中存儲有程序,該程序被處理器執(zhí)行時實現(xiàn)如權(quán)利要求1至8中任一項所述的系統(tǒng)。