Instructions to use hustvl/yolos-tiny with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use hustvl/yolos-tiny with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("object-detection", model="hustvl/yolos-tiny")# Load model directly from transformers import AutoImageProcessor, AutoModelForObjectDetection processor = AutoImageProcessor.from_pretrained("hustvl/yolos-tiny") model = AutoModelForObjectDetection.from_pretrained("hustvl/yolos-tiny") - Notebooks
- Google Colab
- Kaggle
Does it support fine-tunne (custom model training)
#2
by Haonan - opened
Like we are able to fine tune BERT model
Haonan changed discussion title from Does is support fine-tunne (custom model training) to Does it support fine-tunne (custom model training)
Hi,
Yes, check out demo notebooks here: https://github.com/NielsRogge/Transformers-Tutorials/tree/master/YOLOS