[發(fā)明專利]漂浮物識別模型的訓(xùn)練和使用方法、裝置、介質(zhì)及設(shè)備

有效

• 申請?zhí)枺?/td>: 202210127060.4
• 申請日: 2022-02-11
• 公開（公告）號: CN114170483B
• 公開（公告）日: 2022-05-20
• 發(fā)明（設(shè)計）人: 楊帆;朱瑩
• 申請（專利權(quán)）人: 南京甄視智能科技有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/26;G06K9/62
• 代理公司: 北京德崇智捷知識產(chǎn)權(quán)代理有限公司 11467
• 代理人: 季承

本申請公開了一種漂浮物識別模型的訓(xùn)練和使用方法、裝置、介質(zhì)及設(shè)備，屬于圖像處理技術(shù)領(lǐng)域。所述方法包括：獲取訓(xùn)練數(shù)據(jù)集，訓(xùn)練數(shù)據(jù)集中包含多組訓(xùn)練樣本，每組訓(xùn)練樣本中包含一張圖像和對應(yīng)的掩膜圖；利用雙邊高分辨率分割模型中的空間路徑對訓(xùn)練樣本進行處理，得到第一特征圖；利用雙邊高分辨率分割模型中的上下文路徑對訓(xùn)練樣本進行處理，得到四個第二特征圖；根據(jù)四個第二特征圖和第一特征圖進行特征融合和分割解碼，得到四個分支損失函數(shù)；根據(jù)四個分支損失函數(shù)訓(xùn)練雙邊高分辨率分割模型，得到漂浮物識別模型。本申請可以既可以提高漂浮物的檢測效率和準確性，也可以通過多個分支輔助監(jiān)督訓(xùn)練來提高漂浮物識別模型的準確性。

技術(shù)領(lǐng)域

本申請涉及圖像處理技術(shù)領(lǐng)域，特別涉及一種漂浮物識別模型的訓(xùn)練和使用方法、裝置、介質(zhì)及設(shè)備。

背景技術(shù)

河道漂浮物不僅會造成河道污染，還會影響河道的觀賞性。河道漂浮物主要是垃圾（如塑料袋、塑料瓶等）、浮萍、水草等，所以，治理河道污染最有效的方式是及時地處理漂浮物，避免污染面積的擴大，降低對河道中微生物的危害。

當采用人工巡查河道的方式來治理河道污染時，由于人工巡查河道不可避免地受河道長度、視覺盲區(qū)等因素影響，而且人眼有一定的局限性，導(dǎo)致一些漂浮物容易被忽略，因此，亟需一種高效準確的識別河道中漂浮物的方法。

發(fā)明內(nèi)容

本申請?zhí)峁┝艘环N漂浮物識別模型的訓(xùn)練和使用方法、裝置、介質(zhì)及設(shè)備，用于解決人工巡查河道時容易忽略漂浮物的問題。所述技術(shù)方案如下：

第一方面，提供了一種漂浮物識別模型的訓(xùn)練方法，所述方法包括：

獲取訓(xùn)練數(shù)據(jù)集，所述訓(xùn)練數(shù)據(jù)集中包含多組訓(xùn)練樣本，每組訓(xùn)練樣本中包含一張圖像和對應(yīng)的掩膜圖；

利用雙邊高分辨率分割模型中的空間路徑對所述訓(xùn)練樣本進行處理，得到第一特征圖；

利用所述雙邊高分辨率分割模型中的上下文路徑對所述訓(xùn)練樣本進行處理，得到四個第二特征圖；

根據(jù)四個所述第二特征圖和所述第一特征圖進行特征融合和分割解碼，得到四個分支損失函數(shù)；

根據(jù)四個所述分支損失函數(shù)訓(xùn)練所述雙邊高分辨率分割模型，得到漂浮物識別模型。

在一種可能的實現(xiàn)方式中，所述利用所述雙邊高分辨率分割模型中的上下文路徑對所述訓(xùn)練樣本進行處理，得到四個第二特征圖，包括：

當所述雙邊高分辨率分割模型中的上下文路徑中包括串聯(lián)的s個網(wǎng)絡(luò)模塊，且s≥5時，利用所述s個網(wǎng)絡(luò)模塊對所述訓(xùn)練樣本進行處理；

將第s-2個網(wǎng)絡(luò)模塊輸出的特征圖分別確定為第一個第二特征圖和第二個第二特征圖；

對第s-1個網(wǎng)絡(luò)模塊輸出的特征圖進行一次上采樣，將得到的第一上采樣結(jié)果和第s-2個網(wǎng)絡(luò)模塊輸出的特征圖進行融合后確定為第三個第二特征圖；

對第s個網(wǎng)絡(luò)模塊輸出的特征圖進行兩次上采樣，將得到的第二上采樣結(jié)果、所述第一上采樣結(jié)果和第s-2個網(wǎng)絡(luò)模塊輸出的特征圖進行融合后確定為第四個第二特征圖。

在一種可能的實現(xiàn)方式中，所述根據(jù)四個所述第二特征圖和所述第一特征圖進行特征融合和分割解碼，得到四個分支損失函數(shù)，包括：

對所述第一個第二特征圖進行特征融合和分割解碼，得到第一個分支損失函數(shù)；

對所述第二個第二特征圖和所述第一特征圖進行特征融合和分割解碼，得到第二個分支損失函數(shù)；

對所述第三個第二特征圖和所述第一特征圖進行特征融合和分割解碼，得到第三個分支損失函數(shù)；

對所述第四個第二特征圖和所述第一特征圖進行特征融合和分割解碼，得到第四個分支損失函數(shù)。