Image Classification Pipeline
by Community · open-source · Last verified 2026-03-17
End-to-end image classification pipeline that handles dataset loading, preprocessing, model inference, and result export using PyTorch and torchvision. Supports batch inference against any Hugging Face ViT or ResNet checkpoint with configurable confidence thresholds.
https://github.com/huggingface/transformers/tree/main/examples/pytorch/image-classification ↗B
B—Above Average
Adoption: B+Quality: AFreshness: ACitations: BEngagement: F
Specifications
- License
- MIT
- Pricing
- open-source
- Capabilities
- batch-inference, top-k-predictions, confidence-scoring, csv-export
- Integrations
- huggingface, pytorch, wandb
- Use Cases
- product-image-tagging, content-moderation, medical-image-screening
- API Available
- No
- Language
- python
- Dependencies
- torch, torchvision, transformers, pillow, numpy, tqdm
- Environment
- Python 3.10+, CUDA 11.8+
- Est. Runtime
- 2-30 minutes (dataset-dependent)
- Tags
- image-classification, vision, pytorch, torchvision, inference
- Added
- 2026-03-17
- Completeness
- 100%
Index Score
62.7Adoption
78
Quality
80
Freshness
85
Citations
62
Engagement
0