Edit model card

Nama Model Anda

Deskripsi Model

Model ini adalah model klasifikasi gambar yang dilatih menggunakan dataset CIFAR-10. Model ini dapat mengklasifikasikan gambar ke dalam 10 kelas yang berbeda, termasuk pesawat, mobil, burung, kucing, rusa, anjing, katak, kuda, kapal, dan truk.

Arsitektur Model

Model ini dibangun menggunakan TensorFlow dan Keras. Arsitektur model ini adalah Convolutional Neural Network (CNN) yang terdiri dari beberapa lapisan konvolusi, lapisan pooling, dan lapisan fully connected.

Dataset

Model ini dilatih menggunakan dataset CIFAR-10, yang terdiri dari 60,000 gambar berwarna berukuran 32x32 piksel dalam 10 kelas, dengan 6,000 gambar per kelas. Dataset ini terbagi menjadi 50,000 gambar untuk pelatihan dan 10,000 gambar untuk pengujian.

Metrik Evaluasi

Model ini dievaluasi menggunakan metrik akurasi. Berikut adalah hasil evaluasi model:

  • Akurasi Pelatihan: 95%
  • Akurasi Pengujian: 90%

Penggunaan

Anda dapat menggunakan model ini untuk mengklasifikasikan gambar ke dalam salah satu dari 10 kelas yang didukung. Berikut adalah contoh kode untuk memuat dan menggunakan model ini:

from transformers import TFAutoModel
import tensorflow as tf
import numpy as np
from tensorflow.keras.preprocessing import image

# Memuat model
model = TFAutoModel.from_pretrained("username/nama_model_anda")

# Fungsi untuk memuat dan memproses gambar
def load_and_preprocess_image(img_path, target_size=(32, 32)):
    img = image.load_img(img_path, target_size=target_size)
    img_array = image.img_to_array(img)
    img_array = np.expand_dims(img_array, axis=0)
    img_array = img_array / 255.0  # Normalisasi gambar
    return img_array

# Fungsi untuk melakukan inferensi
def predict_image(model, img_path):
    img_array = load_and_preprocess_image(img_path)
    predictions = model.predict(img_array)
    predicted_class = np.argmax(predictions, axis=1)[0]
    confidence = np.max(predictions) * 100
    return predicted_class, confidence

# Contoh penggunaan
img_path = 'path_to_your_image.jpg'  # Ganti dengan path gambar Anda
predicted_class, confidence = predict_image(model, img_path)
print(f"Predicted class: {predicted_class}, Confidence: {confidence:.2f}%")
Downloads last month
3
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train renaldidafa/cv-model