基于遷移學習的中藥飲片圖像分類

摘要： 針對人工識別中藥飲片效率不高、分類具有一定主觀性等問題，提出基于遷移學習的中藥飲片圖像分類方法。自建中藥飲片圖像數(shù)據(jù)集，選取MobileNet-V2模型進行訓練，將在ImageNet數(shù)據(jù)集訓練得到的參數(shù)加載到模型中，修改全連接層節(jié)點數(shù)和SoftMax分類器得到新模型，凍結(jié)除全連接層的參數(shù)，訓練后的新模型準確率達到97.67%。實驗表明，研究的算法可準確對中藥飲片圖像進行分類，而...