Computer Vision Model
This model performs image classification using a pre-trained ResNet model.
Features
- Image classification across 1000 object categories
- Preprocessing included
- Batch processing support
- Configurable confidence threshold
Usage
from image_classifier import ImageClassifier
classifier = ImageClassifier()
result = classifier.classify_image("path/to/image.jpg")
print(result)
Model Details
- Architecture: ResNet-50
- Dataset: ImageNet
- Top-1 Accuracy: 76.2%
Hugging Face Space
This model can be deployed as a Hugging Face Space with a Gradio interface for easy interaction.
Installation
pip install -r requirements.txt
Local Testing
python test_model.py