Python 套件
安裝方式
pip install ponychart-classifier
使用方式
from ponychart_classifier import predict, preload, update
# 預先載入模型(首次呼叫時會自動從遠端下載)
preload()
# 檢查並更新模型至最新版本(比對 ETag,有新版才下載)
update()
# 預測圖片中的角色
result = predict("path/to/image.png")
print(result.labels) # frozenset({'Rarity', 'Fluttershy'})
# 最多標記三個
print(result.rarity) # 0.95 => 模型的信心
print(result.twilight_sparkle) # 0.02 => 低於閾值的不會被標記
現在模型訓練到大概九成以上正確
大概是這樣,給大家玩玩