Image Classification
Transformers
PyTorch
TensorFlow
TensorBoard
Safetensors
timm_wrapper
vision
Generated from Trainer
Instructions to use amyeroberts/vit-base-beans with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use amyeroberts/vit-base-beans with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="amyeroberts/vit-base-beans") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("amyeroberts/vit-base-beans") model = AutoModelForImageClassification.from_pretrained("amyeroberts/vit-base-beans") - Notebooks
- Google Colab
- Kaggle
| { | |
| "architecture": "resnet18", | |
| "data_config": { | |
| "crop_mode": "center", | |
| "crop_pct": 0.95, | |
| "input_size": [ | |
| 3, | |
| 224, | |
| 224 | |
| ], | |
| "interpolation": "bicubic", | |
| "mean": [ | |
| 0.485, | |
| 0.456, | |
| 0.406 | |
| ], | |
| "std": [ | |
| 0.229, | |
| 0.224, | |
| 0.225 | |
| ] | |
| }, | |
| "image_processor_type": "TimmWrapperImageProcessor", | |
| "num_classes": 1000, | |
| "num_features": 512, | |
| "pretrained_cfg": { | |
| "classifier": "fc", | |
| "crop_mode": "center", | |
| "crop_pct": 0.95, | |
| "custom_load": false, | |
| "first_conv": "conv1", | |
| "fixed_input_size": false, | |
| "input_size": [ | |
| 3, | |
| 224, | |
| 224 | |
| ], | |
| "interpolation": "bicubic", | |
| "mean": [ | |
| 0.485, | |
| 0.456, | |
| 0.406 | |
| ], | |
| "num_classes": 1000, | |
| "origin_url": "https://github.com/huggingface/pytorch-image-models", | |
| "paper_ids": "arXiv:2110.00476", | |
| "pool_size": [ | |
| 7, | |
| 7 | |
| ], | |
| "std": [ | |
| 0.229, | |
| 0.224, | |
| 0.225 | |
| ], | |
| "tag": "a1_in1k", | |
| "test_crop_pct": 1.0, | |
| "test_input_size": [ | |
| 3, | |
| 288, | |
| 288 | |
| ] | |
| } | |
| } | |