ONNX 模型库
返回模型

说明文档

Kornia ONNX 仓库

欢迎来到 Kornia ONNX 仓库！本仓库托管了与 Kornia 计算机视觉库相关的模型和算子的 ONNX 文件。在这里，您将找到预训练模型、自定义算子以及相关资源，帮助您使用 ONNX 格式加速机器学习和计算机视觉工作流程。

Kornia ONNX

仓库结构

默认情况下，我们提供带有动态形状的 ONNX 模型。如果模型是以特定输入形状导出的，文件名中会注明，例如 XXX_512x512.onnx。

仓库分为两个主要文件夹：

1. Operators（算子）

此文件夹包含 Kornia 库中使用的各种 ONNX 算子。这些算子设计为模块化，可集成到不同的模型或工作流程中。每个算子都附带简要说明和使用示例。

2. Models（模型）

此文件夹包含使用 Kornia 构建并导出为 ONNX 格式的完整模型。这些模型专为各种计算机视觉任务设计，如图像处理、立体视觉等。每个模型都附有其用途和使用方法的详细说明。

[!Note] 本仓库导出的检测模型不包含后处理器，您可能需要自行处理或通过设置 confidence_filtering=True 重新导出模型。例如，RTDETRDetectorBuilder.to_onnx(confidence_threshold=0.5, confidence_filtering=True)。

快速开始

安装 Kornia 库：

pip install -U kornia

要将多个算子或模型串联成一个计算图，请按以下步骤操作：

import numpy as np
from kornia.onnx import ONNXSequential
# 将来自 HuggingFace 仓库的 ONNX 模型与您自己的本地模型串联在一起
onnx_seq = ONNXSequential(
    "hf://operators/kornia.geometry.transform.flips.Hflip",
    "hf://models/kornia.models.detection.rtdetr_r18vd_640x640",  # 或者您可以使用 "YOUR_OWN_MODEL.onnx"
)
# 准备一些输入数据
input_data = np.random.randn(1, 3, 384, 512).astype(np.float32)
# 执行推理
outputs = onnx_seq(input_data)
# 打印模型输出
print(outputs)

# 导出一个将所有三个模型串联在一起的新 ONNX 模型！
onnx_seq.export("chained_model.onnx")

贡献

我们欢迎您的贡献！如果您想添加自己的 ONNX 算子或模型，欢迎提交 pull request。请确保为任何新增内容包含相关文档和使用示例。

感谢您使用 Kornia ONNX 仓库！欢迎探索并为不断增长的算子和模型集合做出贡献。

kornia/ONNX_models

作者 kornia

↓ 0 ♥ 0

创建时间: 2024-09-06 10:56:00+00:00

更新时间: 2024-09-10 20:10:32+00:00

在 Hugging Face 上查看

文件 (52)

.gitattributes
README.md
banner.jpeg
models/kornia.models.detection.rtdetr_r101vd_480x480.onnx ONNX
models/kornia.models.detection.rtdetr_r101vd_640x640.onnx ONNX
models/kornia.models.detection.rtdetr_r101vd_800x800.onnx ONNX
models/kornia.models.detection.rtdetr_r18vd_480x480.onnx ONNX
models/kornia.models.detection.rtdetr_r18vd_640x640.onnx ONNX
models/kornia.models.detection.rtdetr_r18vd_800x800.onnx ONNX
models/kornia.models.detection.rtdetr_r34vd_480x480.onnx ONNX
models/kornia.models.detection.rtdetr_r34vd_640x640.onnx ONNX
models/kornia.models.detection.rtdetr_r34vd_800x800.onnx ONNX
models/kornia.models.detection.rtdetr_r50vd_480x480.onnx ONNX
models/kornia.models.detection.rtdetr_r50vd_640x640.onnx ONNX
models/kornia.models.detection.rtdetr_r50vd_800x800.onnx ONNX
models/kornia.models.detection.rtdetr_r50vd_m_480x480.onnx ONNX
models/kornia.models.detection.rtdetr_r50vd_m_640x640.onnx ONNX
models/kornia.models.detection.rtdetr_r50vd_m_800x800.onnx ONNX
operators/kornia.color.gray.BgrToGrayscale.onnx ONNX
operators/kornia.color.gray.GrayscaleToRgb.onnx ONNX
operators/kornia.color.gray.RgbToGrayscale.onnx ONNX
operators/kornia.color.hls.HlsToRgb.onnx ONNX
operators/kornia.color.hls.RgbToHls.onnx ONNX
operators/kornia.color.hsv.HsvToRgb.onnx ONNX
operators/kornia.color.hsv.RgbToHsv.onnx ONNX
operators/kornia.color.lab.LabToRgb.onnx ONNX
operators/kornia.color.lab.RgbToLab.onnx ONNX
operators/kornia.color.luv.LuvToRgb.onnx ONNX
operators/kornia.color.luv.RgbToLuv.onnx ONNX
operators/kornia.color.rgb.BgrToRgb.onnx ONNX
operators/kornia.color.rgb.LinearRgbToRgb.onnx ONNX
operators/kornia.color.rgb.RgbToBgr.onnx ONNX
operators/kornia.color.rgb.RgbToLinearRgb.onnx ONNX
operators/kornia.color.sepia.Sepia.onnx ONNX
operators/kornia.color.xyz.RgbToXyz.onnx ONNX
operators/kornia.color.xyz.XyzToRgb.onnx ONNX
operators/kornia.color.ycbcr.RgbToYcbcr.onnx ONNX
operators/kornia.color.ycbcr.YcbcrToRgb.onnx ONNX
operators/kornia.color.yuv.RgbToYuv.onnx ONNX
operators/kornia.color.yuv.YuvToRgb.onnx ONNX
operators/kornia.enhance.normalize.Denormalize_[0.485 0.456 0.406]_[0.229 0.224 0.225].onnx ONNX
operators/kornia.enhance.normalize.Denormalize_[0.5 0.5 0.5]_[0.5 0.5 0.5].onnx ONNX
operators/kornia.enhance.normalize.Normalize_[0.485 0.456 0.406]_[0.229 0.224 0.225].onnx ONNX
operators/kornia.enhance.normalize.Normalize_[0.5 0.5 0.5]_[0.5 0.5 0.5].onnx ONNX
operators/kornia.geometry.transform.affwarp.Resize_256x256.onnx ONNX
operators/kornia.geometry.transform.affwarp.Resize_384x384.onnx ONNX
operators/kornia.geometry.transform.affwarp.Resize_512x512.onnx ONNX
operators/kornia.geometry.transform.affwarp.Resize_576x1024.onnx ONNX
operators/kornia.geometry.transform.affwarp.Resize_640x640.onnx ONNX
operators/kornia.geometry.transform.flips.Hflip.onnx ONNX
operators/kornia.geometry.transform.flips.Rot180.onnx ONNX
operators/kornia.geometry.transform.flips.Vflip.onnx ONNX