usls 0.0.8

A Rust library integrated with ONNXRuntime, providing a collection of ML models.
Documentation

usls

Static Badge Static Badge Static Badge Static Badge
Static Badge Static Badge

A Rust library integrated with ONNXRuntime, providing a collection of Computer Vison and Vision-Language models including YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOv9, YOLOv10, RTDETR, SAM, MobileSAM, EdgeSAM, SAM-HQ, FastSAM, CLIP, BLIP, DINOv2, YOLO-World, PaddleOCR, Depth-Anything and others.

Segment Anything
YOLO + SAM
Monocular Depth Estimation
Panoptic Driving Perception Text-Detection-Recognition

Supported Models

Model Task / Type Example CUDAf32 CUDAf16 TensorRTf32 TensorRTf16
YOLOv5 ClassificationObject DetectionInstance Segmentation demo ✅ ✅ ✅ ✅
YOLOv6 Object Detection demo ✅ ✅ ✅ ✅
YOLOv7 Object Detection demo ✅ ✅ ✅ ✅
YOLOv8 Object DetectionInstance SegmentationClassificationOriented Object DetectionKeypoint Detection demo ✅ ✅ ✅ ✅
YOLOv9 Object Detection demo ✅ ✅ ✅ ✅
YOLOv10 Object Detection demo ✅ ✅ ✅ ✅
RTDETR Object Detection demo ✅ ✅ ✅ ✅
FastSAM Instance Segmentation demo ✅ ✅ ✅ ✅
SAM Segment Anything demo ✅ ✅
MobileSAM Segment Anything demo ✅ ✅
EdgeSAM Segment Anything demo ✅ ✅
SAM-HQ Segment Anything demo ✅ ✅
YOLO-World Object Detection demo ✅ ✅ ✅ ✅
DINOv2 Vision-Self-Supervised demo ✅ ✅ ✅ ✅
CLIP Vision-Language demo ✅ ✅ ✅ visual❌ textual ✅ visual❌ textual
BLIP Vision-Language demo ✅ ✅ ✅ visual❌ textual ✅ visual❌ textual
DB Text Detection demo ✅ ✅ ✅ ✅
SVTR Text Recognition demo ✅ ✅ ✅ ✅
RTMO Keypoint Detection demo ✅ ✅ ❌ ❌
YOLOPv2 Panoptic Driving Perception demo ✅ ✅ ✅ ✅
Depth-Anything(v1, v2) Monocular Depth Estimation demo ✅ ✅ ❌ ❌
MODNet Image Matting demo ✅ ✅ ✅ ✅

Installation

Refer to ort docs

  • Download from ONNXRuntime Releases
  • Then linking 
    export ORT_DYLIB_PATH=/Users/qweasd/Desktop/onnxruntime-osx-arm64-1.17.1/lib/libonnxruntime.1.17.1.dylib

Quick Start

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

Integrate into your own project

# 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???"}