[發明專利]毛絨纖維分類方法及裝置

在審

• 申請號: 202010623714.3
• 申請日: 2020-06-30
• 公開（公告）號: CN111767959A
• 公開（公告）日: 2020-10-13
• 發明（設計）人: 艾國;張帥;張發恩
• 申請（專利權）人: 創新奇智（廣州）科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/46;G06K9/34
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 李飛

1.一種毛絨纖維分類方法，其特征在于，包括：

獲取待分類毛絨纖維圖像；

輸入所述待分類毛絨纖維圖像至預設的特征提取模型，得到對應的特征提取圖像；

輸入所述特征提取圖像至預設的分類模型，得到毛預測幾率及絨預測幾率；

輸入所述特征提取圖像至預設的語義分割模型，計算毛預測比例及絨預測比例；

根據所述毛預測幾率、所述絨預測幾率、所述毛預測比例及所述絨預測比例，得到待分類毛絨纖維的分類結果。

2.根據權利要求1所述的毛絨纖維分類方法，其特征在于，所述根據所述毛預測幾率、所述絨預測幾率、所述毛預測比例及所述絨預測比例，得到待分類毛絨纖維的分類結果，包括：

將所述毛預測幾率與所述毛預測比例相乘，得到毛綜合預測幾率；

將所述絨預測幾率與所述絨預測比例相乘，得到絨綜合預測幾率；

根據所述毛綜合預測幾率和所述絨綜合預測幾率中的較大者，得到待分類毛絨纖維的分類結果。

3.根據權利要求1所述的毛絨纖維分類方法，其特征在于，所述輸入所述特征提取圖像至預設的語義分割模型，計算毛預測比例及絨預測比例，包括：

輸入所述特征提取圖像至預設的語義分割模型，得到對應的背景像素、毛像素及絨像素；

分別統計所述毛像素的個數及所述絨像素的個數；

根據所述毛像素的個數、所述絨像素的個數及所述特征提取圖像的像素總數，計算得到毛預測比例及絨預測比例。

4.根據權利要求1所述的毛絨纖維分類方法，其特征在于，所述預設的分類模型通過以下步驟訓練得到：

獲取分類毛絨纖維掩碼圖像及對應的分類標簽；

以所述分類毛絨纖維掩碼圖像及所述分類標簽對預設的第一神經網絡模型進行訓練，得到所述預設的分類模型。

5.根據權利要求1所述的毛絨纖維分類方法，其特征在于，所述預設的語義分割模型通過以下步驟訓練得到：

獲取分類毛絨纖維掩碼圖像及對應的分類標簽；

以所述分類毛絨纖維掩碼圖像及所述分類標簽對預設的第二神經網絡模型進行訓練，得到所述預設的語義分割模型。

6.一種毛絨纖維分類裝置，其特征在于，包括：

獲取模塊，用于獲取待分類毛絨纖維圖像；

特征提取模塊，用于輸入所述待分類毛絨纖維圖像至預設的特征提取模型，得到對應的特征提取圖像；

第一計算模塊，用于輸入所述特征提取圖像至預設的分類模型，得到毛預測幾率及絨預測幾率；

第二計算模塊，用于輸入所述特征提取圖像至預設的語義分割模型，計算毛預測比例及絨預測比例；

分類模塊，用于根據所述毛預測幾率、所述絨預測幾率、所述毛預測比例及所述絨預測比例，得到待分類毛絨纖維的分類結果。

7.根據權利要求6所述的毛絨纖維分類裝置，其特征在于，所述分類模塊具體用于：

將所述毛預測幾率與所述毛預測比例相乘，得到毛綜合預測幾率；

將所述絨預測幾率與所述絨預測比例相乘，得到絨綜合預測幾率；

根據所述毛綜合預測幾率和所述絨綜合預測幾率中的較大者，得到待分類毛絨纖維的分類結果。

8.根據權利要求6所述的毛絨纖維分類裝置，其特征在于，所述第二計算模塊具體用于：

輸入所述特征提取圖像至預設的語義分割模型，得到對應的背景像素、毛像素及絨像素；

分別統計所述毛像素的個數及所述絨像素的個數；

根據所述毛像素的個數、所述絨像素的個數及所述特征提取圖像的像素總數，計算得到毛預測比例及絨預測比例。

9.一種電子設備，其特征在于，包括存儲器及處理器，所述存儲器用于存儲計算機程序，所述處理器運行所述計算機程序以使所述電子設備執行根據權利要求1至5中任一項所述的毛絨纖維分類方法。

10.一種計算機可讀存儲介質，其特征在于，其存儲有權利要求9所述的電子設備中所使用的計算機程序。