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, }
給大家參考囉
留言板
歡迎留下建議與分享!希望一起交流!感恩!