Saltar al contenido
ModelPublicadoExterno
Browse files

Vitpose Base Simple

with ONNX weights to be compatible with Transformers.js.

Fuentehuggingface.co/onnx-community/vitpose-base-simple/tree/bbebbddd612af2728539103dd5b903a4f5a2da4c
TipoOtherModelo basenielsr/vitpose-base-simpleVersiónRevision bbebbddBcalidad

Fuente del modelo

Tipo
Other
Tamaño del archivo
47,7–328 MB
Modelo base
nielsr/vitpose-base-simple
Versión
vRevision bbebbdd
Actualizado
1 nov 2026
Descargas
396
Fuente
Hugging Face

Descripción de la fuente

https://huggingface.co/nielsr/vitpose-base-simple with ONNX weights to be compatible with Transformers.js.

Usage (Transformers.js)

If you haven't already, you can install the Transformers.js JavaScript library from NPM using:

npm i @huggingface/transformers

Example: Pose estimation w/ onnx-community/vitpose-base-simple.

import { AutoModel, AutoImageProcessor, RawImage } from '@huggingface/transformers';

// Load model and processor
const model_id = 'onnx-community/vitpose-base-simple';
const model = await AutoModel.from_pretrained(model_id);
const processor = await AutoImageProcessor.from_pretrained(model_id);

// Load image and prepare inputs
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/ryan-gosling.jpg';
const image = await RawImage.read(url);
const inputs = await processor(image);

// Predict heatmaps
const { heatmaps } = await model(inputs);

// Post-process heatmaps to get keypoints and scores
const boxes = [[[0, 0, image.width, image.height]]];
const results = processor.post_process_pose_estimation(heatmaps, boxes)[0][0];
console.log(results);

Optionally, visualize the outputs (Node.js usage shown here, using the canvas library):

import { createCanvas, createImageData } from 'canvas';

// Create canvas and draw image
const canvas = createCanvas(image.width, image.height);
const ctx = canvas.getContext('2d');
const imageData = createImageData(image.rgba().data, image.width, image.height);
ctx.putImageData(imageData, 0, 0);

// Draw edges between keypoints
const points = results.keypoints;
ctx.lineWidth = 4;
ctx.strokeStyle = 'blue';
for (const [i, j] of model.config.edges) {
    const [x1, y1] = points[i];
    const [x2, y2] = points[j];
    ctx.beginPath();
    ctx.moveTo(x1, y1);
    ctx.lineTo(x2, y2);
    ctx.stroke();
}

// Draw circle at each keypoint
ctx.fillStyle = 'red';
for (const [x, y] of points) {
    ctx.beginPath();
    ctx.arc(x, y, 8, 0, 2 * Math.PI);
    ctx.fill();
}

// Save image to file
import fs from 'fs';
const out = fs.createWriteStream('pose.png');
const stream = canvas.createPNGStream();
stream.pipe(out)
out.on('finish', () =>  console.log('The PNG file was created.'));
Input imageOutput image
image/jpegimage/png

Note: Having a separate repo for ONNX weights is intended to be a temporary solution until WebML gains more traction. If you would like to make your models web-ready, we recommend converting to ONNX using 🤗 Optimum and structuring your repo like this one (with ONNX weights located in a subfolder named onnx).

Fuentes

1 fuente
Hugging FacePrincipalVersión bbebbddd612af2728539103dd5b903a4f5a2da4c
huggingface.co/onnx-community/vitpose-base-simple/tree/bbebbddd612af2728539103dd5b903a4f5a2da4c

Verificado 1 nov

Download files

Repository files

File inventory only; completeness is unverified.

Artefactos del modelo

8 artefactos

onnx/model_bnb4.onnx

onnx · 49,5 MB · SHA-256 5f6f806c316b…689f · Hugging Face

Download file

onnx/model_fp16.onnx

onnx · FP16 (unverified) · 164 MB · SHA-256 4f8eb8a49cab…1865 · Hugging Face

Download file

onnx/model_int8.onnx

onnx · INT8 (unverified) · 83,4 MB · SHA-256 3a50dd93dc32…6ed5 · Hugging Face

Download file

onnx/model_q4.onnx

onnx · Q4 (unverified) · 54,6 MB · SHA-256 c8ef2bf82220…dd10 · Hugging Face

Download file

onnx/model_q4f16.onnx

onnx · 47,7 MB · SHA-256 e7b533788cdb…2be6 · Hugging Face

Download file

onnx/model_quantized.onnx

onnx · 83,4 MB · SHA-256 688e70371392…6df3 · Hugging Face

Download file

onnx/model_uint8.onnx

onnx · UINT8 (unverified) · 83,4 MB · SHA-256 688e70371392…6df3 · Hugging Face

Download file

onnx/model.onnx

onnx · 328 MB · SHA-256 47af35f9acd4…584f · Hugging Face

Download file

Extractos de fuentes

3 extractos
Readme18 agohuggingface.co/onnx-community/vitpose-base-simple

--- library_name: transformers.js base_model: nielsr/vitpose-base-simple pipeline_tag: keypoint-detection --- https://huggingface.co/nielsr/vitpose-base-simple with ONNX weights to be compatible with Transformers.js. ## Usage (Transformers.js) If you haven't already, you can install the [Transformers.js](https://huggingface.co/docs/transformers.js) JavaScript library from [NPM](https://www.npmjs.com/package/@huggingface/transformers) using: ```bash npm i @huggingface/transformers ``` **Example:** Pose estimation w/ `onnx-community/vitpose-base-simple`. ```js import { AutoModel, AutoImageProcessor, RawImage } from '@huggingface/transformers'; // Load model and processor const model_id = 'onnx-community/vitpose-base-simple'; const model = await AutoModel.from_pretrained(model_id); const processor = await AutoImageProcessor.from_pretrained(model_id); // Load image and prepare inputs const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/ryan-gosling.jpg'; const image = await RawImage.read(url); const inputs = await processor(image); // Predict heatmaps const { heatmaps } = await model(inputs); // Post-process heatmaps to get keypoints and scores const boxes = [[[0, 0, image.width, image.height]]]; const results = processor.post_process_pose_estimation(heatmaps, boxes)[0][0]; console.log(results); ``` Optionally, visualize the outputs (Node.js usage shown here, using the [`canvas`](https://www.npmjs.com/package/canvas) library): ```js import { createCanvas, createImageData } from 'canvas'; // Create canvas and draw image const canvas = createCanvas(image.width, image.height); const ctx = canvas.getContext('2d'); const imageData = createImageData(image.rgba().data, image.width, image.height); ctx.putImageData(imageData, 0, 0); // Draw edges between keypoints const points = results.keypoints; ctx.lineWidth = 4; ctx.strokeStyle = 'blue'; for (const [i, j] of model.config.edges) { const [x1, y1] = points[i]; const [x2, y2] = points[j]; ctx.beginPath(); ctx.moveTo(x1, y1); ctx.lineTo(x2, y2); ctx.stroke(); } // Draw circle at each keypoint ctx.fillStyle = 'red'; for (const [x, y] of points) { ctx.beginPath(); ctx.arc(x, y, 8, 0, 2 * Math.PI); ctx.fill(); } // Save image to file import fs from 'fs'; const out = fs.createWriteStream('pose.png'); const stream = canvas.createPNGStream(); stream.pipe(out) out.on('fin...

Note1 novhuggingface.co/onnx-community/vitpose-base-simple

Source context: 396 downloads · 3 likes · Pipeline keypoint-detection · Library transformers.js · Repo onnx-community/vitpose-base-simple

Note18 agohuggingface.co/onnx-community/vitpose-base-simple

Source context: 228 downloads · 3 likes · Pipeline keypoint-detection · Library transformers.js · Repo onnx-community/vitpose-base-simple