Instructions to use hustvl/yolos-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use hustvl/yolos-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("object-detection", model="hustvl/yolos-base")# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForObjectDetection processor = AutoImageProcessor.from_pretrained("hustvl/yolos-base") model = AutoModelForObjectDetection.from_pretrained("hustvl/yolos-base", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Download preprocessor_config.json from hustvl/yolos-base: direct link, hf CLI and curl.
- Browser
- Download file 296 Bytes
-
https://huggingface.co/hustvl/yolos-base/resolve/main/preprocessor_config.json
- Command line
-
hf download hf://hustvl/yolos-base/preprocessor_config.json
-
curl -L -o preprocessor_config.json https://huggingface.co/hustvl/yolos-base/resolve/main/preprocessor_config.json
296 Bytes
| { | |
| "do_normalize": true, | |
| "do_resize": true, | |
| "feature_extractor_type": "YolosFeatureExtractor", | |
| "format": "coco_detection", | |
| "image_mean": [ | |
| 0.485, | |
| 0.456, | |
| 0.406 | |
| ], | |
| "image_std": [ | |
| 0.229, | |
| 0.224, | |
| 0.225 | |
| ], | |
| "size": {"shortest_edge": 800, "longest_edge": 1333} | |
| } | |