Downloads · 30 days
8
18% of all-time downloads
vit-beans-leaf-disease
YMmim/vit-beans-leaf-disease
vit-beans-leaf-disease is a image classification model from YMmim. Use it when you need a label for an image. The card lists the license as apache-2.0.
google/vit-base-patch16-224-in21k를 beans 데이터셋으로 미세조정한 이미지 분류 모델입니다. 콩잎 사진을 3개 클래스로 분류합니다.
All-time downloads
44
Public
Parameters
85.8M
686 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors343 MB · 100%
At a glance
- Task
- Image Classification
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Aug 21, 2026
- Updated
- Aug 21, 2026
- SHA
- 098f8f42
Base models
Datasets
Model card
From the Hugging Face model README
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")
- Task
- Image Classification
- Type
- vit
- License
- apache-2.0
- Created
- Aug 21, 2026
- Updated
- Aug 21, 2026