# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("JustFadjrin/batik-vit-model-classification")
model = AutoModelForImageClassification.from_pretrained("JustFadjrin/batik-vit-model-classification")Quick Links
Batik ViT Model Confident
Model ini adalah model klasifikasi jenis batik berbasis Vision Transformer (ViT).
Model digunakan pada project Batik ViT Classifier untuk mengklasifikasikan gambar batik Indonesia.
Dataset
Dataset berasal dari Kaggle:
Batik Nusantara (Batik Indonesia) Dataset
Author: HendryHB
Link: https://www.kaggle.com/datasets/hendryhb/batik-nusantara-batik-indonesia-dataset
Jenis Batik yang Dapat Dideteksi
Model ini dilatih untuk mengenali 20 jenis batik:
- Aceh Pintu Aceh
- Bali Barong
- Bali Merak
- DKI Ondel Ondel
- Jawa Barat Megamendung
- Jawa Timur Pring
- Kalimantan Dayak
- Lampung Gajah
- Madura Mataketeran
- Maluku Pala
- NTB Lumbung
- Papua Asmat
- Papua Cendrawasih
- Papua Tifa
- Solo Parang
- Sulawesi Selatan Lontara
- Sumatera Barat Rumah Minang
- Sumatera Utara Boraspati
- Yogyakarta Kawung
- Yogyakarta Parang
Catatan
Model ini tidak 100% akurat karena dataset masih terbatas dan jenis batik di Indonesia sangat banyak. Hasil prediksi dipengaruhi oleh kualitas gambar, pencahayaan, sudut pengambilan foto, background, dan kemiripan motif antar kelas.
- Downloads last month
- 28
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="JustFadjrin/batik-vit-model-classification") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")