How to use AdamCodd/yolos-small-person with Transformers.js:
// npm i @huggingface/transformers import { pipeline } from '@huggingface/transformers'; // Allocate pipeline const pipe = await pipeline('object-detection', 'AdamCodd/yolos-small-person');
How to use AdamCodd/yolos-small-person with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("object-detection", model="AdamCodd/yolos-small-person")
# Load model directly from transformers import AutoImageProcessor, AutoModelForObjectDetection processor = AutoImageProcessor.from_pretrained("AdamCodd/yolos-small-person") model = AutoModelForObjectDetection.from_pretrained("AdamCodd/yolos-small-person", device_map="auto")
The community tab is the place to discuss and collaborate with the HF community!
