rishitdagli/cppe-5
Viewer • Updated • 1.03k • 1.53k • 23
How to use emre/detr-resnet-50_finetuned_cppe5 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("object-detection", model="emre/detr-resnet-50_finetuned_cppe5") # Load model directly
from transformers import AutoImageProcessor, AutoModelForObjectDetection
processor = AutoImageProcessor.from_pretrained("emre/detr-resnet-50_finetuned_cppe5")
model = AutoModelForObjectDetection.from_pretrained("emre/detr-resnet-50_finetuned_cppe5")This model is a fine-tuned version of facebook/detr-resnet-50 on the cppe-5 dataset.
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
Step Training Loss
300 2.162200
600 2.011000
1200 1.779500
Base model
facebook/detr-resnet-50