ONNX 模型库
返回模型

说明文档

https://huggingface.co/facebook/hiera-base-224-in1k-hf 的 ONNX 权重,用于兼容 Transformers.js。

用法 (Transformers.js)

如果你还没有安装,可以通过 NPM 安装 Transformers.js JavaScript 库:

npm i @huggingface/transformers

示例: 使用 onnx-community/hiera-base-224-in1k-hf 进行图像分类

import { pipeline } from '@huggingface/transformers';

// 创建图像分类流水线
const classifier = await pipeline('image-classification', 'onnx-community/hiera-base-224-in1k-hf');

// 对图像进行分类
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/tiger.jpg';
const output = await classifier(url);
console.log(output);
// [
//   { label: 'tiger, Panthera tigris', score: 0.759323239326477 },
//   { label: 'tiger cat', score: 0.08907650411128998 },
//   { label: 'lynx, catamount', score: 0.0008640264859423041 },
//   { label: 'jaguar, panther, Panthera onca, Felis onca', score: 0.0007982379174791276 },
//   { label: 'leopard, Panthera pardus', score: 0.00041627752943895757 }
// ]

注意:为 ONNX 权重创建单独的仓库是一个临时方案,直到 WebML 获得更广泛的支持。如果你想让你的模型支持 Web 端使用,我们建议使用 🤗 Optimum 转换为 ONNX 格式,并按照此仓库的结构组织(将 ONNX 权重放在名为 onnx 的子文件夹中)。

onnx-community/hiera-base-224-in1k-hf

作者 onnx-community

image-classification transformers.js
↓ 1 ♥ 0

创建时间: 2024-08-29 01:56:11+00:00

更新时间: 2024-10-08 13:54:23+00:00

在 Hugging Face 上查看

文件 (13)

.gitattributes
README.md
config.json
onnx/model.onnx ONNX
onnx/model_bnb4.onnx ONNX
onnx/model_fp16.onnx ONNX
onnx/model_int8.onnx ONNX
onnx/model_q4.onnx ONNX
onnx/model_q4f16.onnx ONNX
onnx/model_quantized.onnx ONNX
onnx/model_uint8.onnx ONNX
preprocessor_config.json
quantize_config.json