[發明專利]一種基于注意力機制的沙粒圖像分類方法

在審

• 申請號: 202010114805.4
• 申請日: 2020-02-25
• 公開（公告）號: CN111382676A
• 公開（公告）日: 2020-07-07
• 發明（設計）人: 葛士平;顧慶;蔣智威;郝慧珍;董小龍;胡修棉
• 申請（專利權）人: 南京大學
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06N3/04;G06N3/08
• 代理公司: 江蘇圣典律師事務所 32237
• 代理人: 賀翔

1.一種基于注意力機制的沙粒圖像分類方法，其特征在于，包括步驟如下：

1）收集已標注沙粒圖像構建訓練集；

2）對沙粒圖像進行預處理；

3）設計基于注意力機制的卷積網絡結構；

4）訓練沙粒圖像自動分類模型；

5）使用上述沙粒圖像自動分類模型預測沙粒圖像的類別。

2.根據權利要求1所述的基于注意力機制的沙粒圖像分類方法，其特征在于，所述步驟1）具體包括：針對不同河流沙類別，包括：單晶石英、多晶石英、斜長石、堿性長石、火山巖、侵入巖、頁巖、板巖、片麻巖；收集每一個類別的沙粒圖像，基于這些圖像構建訓練集，包括沙粒圖像集和對應的沙粒類別標注集。

3.根據權利要求1所述的基于注意力機制的沙粒圖像分類方法，其特征在于，所述步驟2）具體包括：將每張沙粒圖像具有統一的張量格式，用于卷積網絡模型的輸入。

4.根據權利要求1所述的基于注意力機制的沙粒圖像分類方法，其特征在于，所述步驟3）中的卷積網絡結包括：基礎網絡框架和注意力模塊。

5.根據權利要求1所述的基于注意力機制的沙粒圖像分類方法，其特征在于，所述步驟4）具體包括：基于訓練集和損失函數定義，利用已標注沙粒圖像完成基于注意力機制的卷積網絡訓練，得到端到端的沙粒圖像多類別自動分類模型。

6.根據權利要求1所述的基于注意力機制的沙粒圖像分類方法，其特征在于，所述步驟5）具體包括：輸入顯微鏡下拍照獲得的沙粒圖像，即新樣本數據，使用訓練后的模型對新樣本數據進行預測，輸出沙粒圖像所屬類別及置信度。