煙火檢測模型的訓練方法、煙火檢測方法及設備

基本信息

申請?zhí)?/td> CN202110215838.2 申請日 -
公開（公告）號 CN112949453A 公開（公告）日 2021-06-11
申請公布號 CN112949453A 申請公布日 2021-06-11
分類號 G06K9/00;G06K9/62;G06N3/04;G06N3/08 分類 計算；推算；計數;
發(fā)明人 曹毅超;孫飛;施燕平;李溯;陳斌鋒;封曉強 申請（專利權）人 南京恩博科技有限公司
代理機構 江蘇瑞途律師事務所 代理人 蔣海軍
地址 210000 江蘇省南京市玄武區(qū)長江后街6號東大科技園創(chuàng)業(yè)園長江樓B208室
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種煙火檢測模型的訓練方法、煙火檢測方法及設備，屬于煙火檢測技術領域。訓練方法包括：構建視頻煙火樣本數據集；將輸入圖像輸入紋理支路進行特征提取，得到多尺度特征表示，并通過特征金字塔將多尺度特征表示融合為紋理特征；計算輸入圖像與參考圖像的幀差圖像，將幀差圖像輸入運動支路計算運動注意力權重圖；對紋理特征進行運動感知增強；利用弱注意力模塊生成弱煙火目標掩碼；根據運動感知增強后的紋理特征和弱煙火目標掩碼，得到煙火特征圖，并進行煙火目標檢測。并且提出了利用所訓練出的模型進行煙火檢測的方法，以及執(zhí)行所述煙火檢測方法的設備。本發(fā)明能夠有效地提高煙火的檢測準確率，并且計算成本低，便于部署。