需同意授權

茶樹病害影像辨識模型

茶樹病害影像辨識模型

茶樹病害影像辨識模型可將田間拍照影像到的茶樹病害徵狀透過此模型作快速辨識。

AI模型訓練過程

團隊利用FastAI開發茶樹的病害影響辨識模型,並使用ResNet-34模型架構為基礎進行訓練,通過添加自定義的多標籤分類指標、學習率查找和微調技術來優化模型的訓練過程。 所使用的資料集為自行蒐集的實際田間徵狀照片,並透過專家進行人工標註徵狀位置。 運用該資料集進行訓練之前,有進行資料品質監控(Open CV和AutoEncoder)。

訓練所使用資料集

Tea plant nutrient, disease and pest symptom image dataset (茶樹營養,病蟲害徵狀影像資料集)

目前可辨識之茶樹病害類型

赤葉枯病、餅病、髮狀病、輪斑病、煤煙病、藻斑病、地衣、枝枯病、藻類。

模型使用步驟 - 請下載Tea_Disease_v02檔案

  1. 使用anaconda建立fastai虛擬環境
  2. 根據安裝的cuda版本下載pytorch,網址:https://pytorch.org/get-started/locally/
  3. 下載模型壓縮檔,並解壓縮到一資料夾中
  4. 將anaconda終端所在位置移至解壓縮資料夾
  5. 安裝Python所需要的庫 (requirement.txt)
  6. 將想識別的茶樹照片放置到source資料夾內
  7. 運行腳本 python TeaDiseaseAI_Test.py
  8. 判讀結果的照片自動保存至result資料夾,該照片會顯示判識的茶樹病害名稱

下圖即為測試照片原圖與此模型所辨識的病害結果、Grad-cam繪製模型注意力的熱力圖。

茶樹病害影像辨識模型可視化結果

Data and Resources

Additional Info

Field Value
Last Updated August 20, 2024, 15:57 (+0800)
Created June 27, 2023, 11:23 (+0800)

Recommended Dataset