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 ↗C
C—Below Average
Adoption: B+Quality: AFreshness: ACitations: FEngagement: 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
- 80%
Index Score
47Adoption
78
Quality
80
Freshness
85
Citations
0
Engagement
0