Xenova/detr-resnet-50
Xenova
Detección de objetos
Modelo para la detección de objetos utilizando Transformers.js con pesos ONNX para ser compatible con Transformers.js.
Como usar
Si aún no lo has hecho, puedes instalar la biblioteca JavaScript Transformers.js desde NPM usando:
npm i @xenova/transformers
Ejemplo: Realiza la detección de objetos con Xenova/detr-resnet-50.
import { pipeline } from '@xenova/transformers';
const detector = await pipeline('object-detection', 'Xenova/detr-resnet-50');
const img = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/cats.jpg';
const output = await detector(img, { threshold: 0.9 });
// [{
// "score": 0.9976370930671692,
// "label": "remote",
// "box": { "xmin": 31, "ymin": 68, "xmax": 190, "ymax": 118 }
// },
// ...
// {
// "score": 0.9984092116355896,
// "label": "cat",
// "box": { "xmin": 331, "ymin": 19, "xmax": 649, "ymax": 371 }
// }]
Funcionalidades
- Detección de objetos
- Compatible con Transformers.js
- Pesos ONNX
Casos de uso
- Detección de objetos en imágenes
- Creación de demostraciones de detección de objetos