rishitdagli/cppe-5
Viewer โข Updated โข 1.03k โข 1.6k โข 25
How to use Aryan-401/detr-resnet-50-cppe5 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("object-detection", model="Aryan-401/detr-resnet-50-cppe5") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoImageProcessor, AutoModelForObjectDetection
processor = AutoImageProcessor.from_pretrained("Aryan-401/detr-resnet-50-cppe5")
model = AutoModelForObjectDetection.from_pretrained("Aryan-401/detr-resnet-50-cppe5", device_map="auto")# pip install -U transformers accelerate
# Load model directly
from transformers import AutoImageProcessor, AutoModelForObjectDetection
processor = AutoImageProcessor.from_pretrained("Aryan-401/detr-resnet-50-cppe5")
model = AutoModelForObjectDetection.from_pretrained("Aryan-401/detr-resnet-50-cppe5", device_map="auto")
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("object-detection", model="Aryan-401/detr-resnet-50-cppe5")