8 个版本

0.0.10	2024 年 8 月 11 日
0.0.9	2024 年 8 月 3 日
0.0.7	2024 年 7 月 31 日
0.0.2	2024 年 3 月 30 日

190 在机器学习中排名

每月 602 次下载

MIT 许可证

365KB
10K SLoC

usls

一个集成 ONNXRuntime 的 Rust 库，提供一系列包括 YOLOv5、YOLOv6、YOLOv7、YOLOv8、YOLOv9、YOLOv10、RTDETR、SAM、MobileSAM、EdgeSAM、SAM-HQ、FastSAM、CLIP、BLIP、DINOv2、YOLO-World、PaddleOCR、Depth-Anything、GroundingDINO 等在内的 计算机视觉 和 视觉-语言 模型。

Segment Anything

YOLO + SAM

单目深度估计

全景驾驶感知	文本检测-识别

支持模型

模型	任务 / 类型	示例	CUDA f32	CUDA f16	TensorRT f32	TensorRT f16
YOLOv5	分类目标检测实例分割	demo	✅	✅	✅	✅
YOLOv6	目标检测	demo	✅	✅	✅	✅
YOLOv7	目标检测	demo	✅	✅	✅	✅
YOLOv8	目标检测实例分割分类方向性目标检测关键点检测	demo	✅	✅	✅	✅
YOLOv9	目标检测	demo	✅	✅	✅	✅
YOLOv10	目标检测	demo	✅	✅	✅	✅
RTDETR	目标检测	demo	✅	✅	✅	✅
FastSAM	实例分割	demo	✅	✅	✅	✅
SAM	Segment Anything	demo	✅	✅
MobileSAM	Segment Anything	demo	✅	✅
EdgeSAM	Segment Anything	demo	✅	✅
SAM-HQ	Segment Anything	demo	✅	✅
YOLO-World	目标检测	demo	✅	✅	✅	✅
DINOv2	视觉自监督	demo	✅	✅	✅	✅
CLIP	视觉-语言	demo	✅	✅	✅ 视觉 ❌ 文本	✅ 视觉 ❌ 文本
BLIP	视觉-语言	demo	✅	✅	✅ 视觉 ❌ 文本	✅ 视觉 ❌ 文本
DB	文本检测	demo	✅	✅	✅	✅
SVTR	文本识别	demo	✅	✅	✅	✅
RTMO	关键点检测	demo	✅	✅	❌	❌
YOLOPv2	全景驾驶感知	demo	✅	✅	✅	✅
Depth-Anything (v1, v2)	单目深度估计	demo	✅	✅	❌	❌
MODNet	图像去背	demo	✅	✅	✅	✅
GroundingDINO	基于语言的开放集检测	demo	✅	✅

安装

参考 ort 文档

对于 Linux 或 MacOS 用户

从 ONNXRuntime 版本发布下载

然后链接

export ORT_DYLIB_PATH=/Users/qweasd/Desktop/onnxruntime-osx-arm64-1.17.1/lib/libonnxruntime.1.17.1.dylib

快速开始

cargo run -r --example yolo   # blip, clip, yolop, svtr, db, ...

将其集成到您自己的项目中

# Add `usls` as a dependency to your project's `Cargo.toml`
cargo add usls

# Or you can use specific commit
usls = { git = "https://github.com/jamjamjon/usls", rev = "???sha???"}

依赖项

~35–47MB
~798K SLoC