How to use gghsgn/safety_helmet_detection with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("object-detection", model="gghsgn/safety_helmet_detection")
# Load model directly from transformers import AutoImageProcessor, AutoModelForObjectDetection processor = AutoImageProcessor.from_pretrained("gghsgn/safety_helmet_detection") model = AutoModelForObjectDetection.from_pretrained("gghsgn/safety_helmet_detection", device_map="auto")
Umsh