rishitdagli/cppe-5
Viewer โข Updated โข 1.03k โข 1.56k โข 25
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") # pip install -U transformers accelerate
# 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", device_map="auto")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