rishitdagli/cppe-5
Viewer • Updated • 1.03k • 2.71k • 22
How to use Yorai/yolos-tiny_finetuned_cppe-5 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("object-detection", model="Yorai/yolos-tiny_finetuned_cppe-5") # Load model directly
from transformers import AutoImageProcessor, AutoModelForObjectDetection
processor = AutoImageProcessor.from_pretrained("Yorai/yolos-tiny_finetuned_cppe-5")
model = AutoModelForObjectDetection.from_pretrained("Yorai/yolos-tiny_finetuned_cppe-5")This model is a fine-tuned version of hustvl/yolos-tiny on the cppe-5 dataset.
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
Base model
hustvl/yolos-tiny