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

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