fastai 很常常拿來比賽用

因為它把分多東西都包套起來使用

預設都會用

但缺點是有些東西要知道目前怎麼做

或者是需要調整就稍微困難一點

例如我用資料夾的方式讀取圖片進入

標籤其實是給資料夾名稱

但是我不知道中文會對應什麼編號?

因為fastai 會將標籤用 onehot encode 變成編號

但我就不知道資料夾名稱對應的分類編號了

找一找可以用以下方法

 

dl_df = learn.dls.test_dl(df, bs=cfg['bs']*2)
class_indices = learn.dls.vocab.o2i

 

print 會顯示

 

class_indices =  {'菊花': 0, '早運草': 1, '杜仲': 2, }

給大家參考囉