[發明專利]一種船舶檢測方法、系統及計算機存儲介質

有效

• 申請號: 202310209497.7
• 申請日: 2023-03-07
• 公開（公告）號: CN116052094B
• 公開（公告）日: 2023-06-09
• 發明（設計）人: 陳江海;魯杰;曹彩霞;吳顯德;張進虎
• 申請（專利權）人: 浙江華是科技股份有限公司
• 主分類號: G06V20/54
• 分類號: G06V20/54;G06V10/72;G06V10/774;G06V10/766;G06V10/26;G06V10/80
• 代理公司: 北京城烽知識產權代理事務所(特殊普通合伙) 11829
• 代理人: 王新月

本發明公開一種船舶檢測方法、系統及計算機存儲介質。其中，該方法包括：將原始訓練集進行復制，并將復制好的圖片從不同種類的數據增強中隨機選擇一種進行數據增強，將上述圖片進行模型訓練，得到第一更新船舶檢測模型；對原始訓練集進行不同種類的數據增強，得到增強訓練集；將原始訓練集、增強訓練集通過第一更新船舶檢測模型進行訓練，計算得到全部目標的差異值，根據差異值將目標進行裁剪并粘貼到對應的原始圖片中并進行融合操作，以使目標與原始圖片的邊緣虛化，中心突出；本發明利用原始迭代圖片對增強迭代圖片的引導，使模型不僅能學習原始迭代圖片特征，還能學習各種場景下的增強迭代圖片特征，提高不同環境下的準確率和降低錯誤率。

技術領域

本發明涉及目標檢測技術領域，具體而言，涉及一種船舶檢測方法、系統及計算機存儲介質。

背景技術

隨著水上交通的不斷發展，導致水上交通指揮工作人員的作業量急劇上升。其中船舶檢查是水上交通的重要工作之一，為了減少水上交通工作人員監視船舶的工作量，船舶檢測在水上交通被廣泛使用。但水上交通環境背景復雜且受到白天晚上光照影響，以及人為因素在訓練集標定所犯的人為錯誤，目前的船舶檢測方法在船舶檢測中存在以下問題：

（1）水上交通復雜，數據采集無法全面采集，對環境擬合存在問題，誤檢偏多；

（2）水上交通實時監測，受到一年四季和一天光線變換影響；

（3）拷貝粘貼的目標存在比較明顯的圖像分界線；

（4）因人工出現錯誤標定結果。

針對上述問題，目前尚未提出有效的解決方案。

發明內容

本發明實施例中提供一種船舶檢測方法、系統及計算機存儲介質，以解決現有技術中水上交通復雜，數據采集無法全面采集，對環境擬合存在問題，誤檢偏多；水上交通實時監測，受到一年四季和一天光線變換影響；拷貝粘貼的目標存在比較明顯的圖像分界線；因人工出現錯誤標定結果的問題。

為達到上述目的，一方面，本發明提供了一種船舶檢測方法，其中，該方法包括：S1，從原始訓練集中抽取原始迭代圖片；對每張原始迭代圖片分別進行復制，將復制好的每張原始迭代圖片從不同種類的數據增強中隨機選擇一種進行數據增強，得到增強迭代圖片；S2，將每張原始迭代圖片輸入到所述初始船舶檢測模型中進行模型訓練，得到原始目標預測結果；以及將每張增強迭代圖片輸入到所述初始船舶檢測模型中進行模型訓練，得到增強目標預測結果；根據所述原始目標預測結果和所述增強目標預測結果計算得到迭代總損失值；根據所述迭代總損失值對所述初始船舶檢測模型進行更新，得到當前迭代船舶檢測模型；將原始訓練集全部圖片訓練完，得到第一更新船舶檢測模型以及第一輪總損失值；S3，將原始訓練集輸入到所述第一更新船舶檢測模型中進行模型訓練，得到每個目標的原始損失值；將原始訓練集中每張圖片分別進行不同種類的數據增強得到增強訓練集，將所述增強訓練集輸入到所述第一更新船舶檢測模型中進行模型訓練，得到每個目標的增強損失值；根據所述原始損失值、所述增強損失值計算得到增強訓練集中每個目標的差異值；將全部的所述差異值從大到小排列，并將排列后的第一預設范圍內的差異值對應的目標進行裁剪；將裁剪的目標粘貼到對應的原始圖片中并進行融合操作，以使目標與所述原始圖片的邊緣虛化，中心突出，得到第一粘貼圖片集；S4，將所述第一粘貼圖片集進行S1、S2操作，得到第二更新船舶檢測模型以及第二輪總損失值；S5，將所述原始訓練集輸入到所述第二更新船舶檢測模型，重復S3操作，得到第二粘貼圖片集；S6，重復所述S4、S5，直至第二輪總損失值在第二預設范圍內波動，停止模型訓練，得到目標船舶檢測模型；S7，將待檢測圖片輸入到所述目標船舶檢測模型中進行檢測，得到目標位置和類別。

可選的，所述融合操作通過以下公式計算：

；

；

；

；