File size: 708 Bytes
c4897b6 212e7ed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
---
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}")
```
|