YMmim/vit-beans-leaf-disease

ViT 콩잎 병충해 분류 (beans, 3-class)

google/vit-base-patch16-224-in21k를 beans 데이터셋으로 미세조정한 이미지 분류 모델입니다. 콩잎 사진을 3개 클래스로 분류합니다.

라벨

id	클래스	뜻
0	angular_leaf_spot	각반병
1	bean_rust	녹병
2	healthy	정상

성능 (test 세트)

• Accuracy: 0.969
• F1 (macro): 0.969

학습 정보

• Base: google/vit-base-patch16-224-in21k
• 데이터: beans (train 1034 / val 133 / test 128)
• 5 epochs, batch 16, lr 5e-5, warmup 0.1, fp16, RTX 3060
• transformers 4.45

사용법

from transformers import pipeline
clf = pipeline("image-classification", model="YMmim/vit-beans-leaf-disease")
clf("leaf.jpg")