|
|
--- |
|
|
tags: |
|
|
- image-classification |
|
|
pipeline_tag: image-classification |
|
|
--- |
|
|
|
|
|
# Garbage Classification Model |
|
|
|
|
|
A ResNet50 model fine-tuned on the Garbage Classification dataset. |
|
|
|
|
|
## Model Details |
|
|
|
|
|
- Base Architecture: ResNet50 |
|
|
- Input Size: 224x224 |
|
|
- Classes: battery, biological, cardboard, clothes, glass, metal, paper, plastic, shoes, trash |
|
|
|
|
|
## Usage |
|
|
|
|
|
```python |
|
|
from transformers import pipeline |
|
|
from PIL import Image |
|
|
|
|
|
# Create pipeline |
|
|
classifier = pipeline("image-classification", model="YOUR_USERNAME/garbage-classification") |
|
|
|
|
|
# Load image |
|
|
image = Image.open("test_image.jpg") |
|
|
|
|
|
# Make prediction |
|
|
result = classifier(image) |
|
|
print(f"Prediction: {result[0]['label']}, Confidence: {result[0]['score']:.2f}") |
|
|
``` |
|
|
|
