[發(fā)明專利]一種皮膚病圖像分類方法

在審

• 申請?zhí)枺?/td>: 201810438389.6
• 申請日: 2018-05-09
• 公開（公告）號: CN108596273A
• 公開（公告）日: 2018-09-28
• 發(fā)明（設(shè)計(jì)）人: 宋艷枝;郭松陶
• 申請（專利權(quán)）人: 合肥黎曼信息科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62
• 代理公司: 上海精晟知識(shí)產(chǎn)權(quán)代理有限公司 31253
• 代理人: 馮子玲

1.一種皮膚病圖像分類方法，其特征在于，包括如下步驟：

步驟S01選取圖像預(yù)處理方式：針對皮膚病圖像具體分析，選取預(yù)處理方式；

步驟S02選擇組件分類器：選擇深度卷積神經(jīng)網(wǎng)絡(luò)作為組件分類器，設(shè)計(jì)整體模型結(jié)構(gòu)；

步驟S03選擇訓(xùn)練方式和整體模型結(jié)構(gòu)：根據(jù)預(yù)處理方式的復(fù)雜程度及計(jì)算機(jī)的承受能力，確定訓(xùn)練模型的方式；

步驟S04模型訓(xùn)練：使用訓(xùn)練集訓(xùn)練端對端模型；

步驟S05新樣本預(yù)測：使用訓(xùn)練好的模型對測試集進(jìn)行預(yù)測，完成圖像分類任務(wù)。

2.根據(jù)權(quán)利要求1所述的一種皮膚病圖像分類方法，其特征在于，所述步驟S01中，預(yù)處理方式包括：圖像顏色平衡、去除毛發(fā)和疾病區(qū)域檢測；

其中，所述圖像顏色平衡通過simplest color balance或gray world算法，用于提升圖片質(zhì)量；所述去除毛發(fā)通過掩膜重建算法，用于將圖片進(jìn)行降噪；所述疾病區(qū)域檢測通過overfeat或RCNN算法，用于統(tǒng)一皮膚病圖片的尺寸規(guī)則。

3.根據(jù)權(quán)利要求1所述的一種皮膚病圖像分類方法，其特征在于，所述步驟S03中，通過組件分類器提取圖像特征，再進(jìn)行拼接或其他組合；所述組件分類器為深度卷積網(wǎng)絡(luò)ResNet。

4.根據(jù)權(quán)利要求1所述的一種皮膚病圖像分類方法，其特征在于，所述步驟S04中，模型訓(xùn)練方法包括：

方法一：

步驟S11經(jīng)過不同預(yù)處理的圖像訓(xùn)練組件ResNet，標(biāo)簽與原圖像相同；

步驟S12獲取各個(gè)組件ResNet的特征層拼接到一起，對應(yīng)原圖像的標(biāo)簽；

步驟S13添加若干層全連接層，訓(xùn)練人工神經(jīng)網(wǎng)絡(luò)；

步驟S14根據(jù)整體模型分類表現(xiàn)，交替調(diào)整特征層和人工神經(jīng)網(wǎng)絡(luò)；

步驟S15固定各部分的參數(shù)，將整體作為預(yù)測模型；

方法二：

步驟S21將組件ResNet的輸出層去掉，并將特征層拼接在一起；

步驟S22再添加若干層全連接層；

步驟S23添加一個(gè)輸出層；

步驟S24將模型作為一個(gè)整體來訓(xùn)練，并直接實(shí)現(xiàn)端到端的預(yù)測。