object_detector 0.2.1

Object detection using ORT and the yoloe-26-seg model. This model can detect multiple objects per image, each having a tag, pixel-level mask, and a boundingbox. It's pretrained, it has a vocabulary of 4000+ objects.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

[package]
name = "object_detector"
description = "Object detection using ORT and the yoloe-26-seg model. This model can detect multiple objects per image, each having a tag, pixel-level mask, and a boundingbox. It's pretrained, it has a vocabulary of 4000+ objects. "
version = "0.2.1"
edition = "2024"
license = "AGPL-3.0"
homepage = "https://github.com/RuurdBijlsma/object_detector-rs"
repository = "https://github.com/RuurdBijlsma/object_detector-rs"
documentation = "https://docs.rs/object_detector"
exclude = ["assets", ".github", "TODO.md", "justfile", "output", "py-yolo"]

[dependencies]
ort = { version = "2.0.0-rc.12", features = ["ndarray"] }
ndarray = "0.17.2"
image = "0.25.10"
serde_json = "1.0.149"
num_cpus = "1.17.0"
rayon = "1.11.0"
thiserror = "2.0.18"
serde = { version = "1.0.228", optional = true, features = ["derive"] }
hf-hub = { version = "0.5.0", features = ["tokio"], optional = true }
bon = "3.9.1"
open_clip_inference = { version = "0.3.0" }

[dev-dependencies]
serde = { version = "1.0", features = ["derive"] }
criterion = { version = "0.8.2", features = ["html_reports"] }
imageproc = "0.26.1"
ab_glyph = "0.2"
tokio = { version = "1.50", features = ["full"] }
color-eyre = "0.6.5"

[features]
default = ["download-binaries", "copy-dylibs", "hf-hub", "serde"]
serde = ["dep:serde"]
hf-hub = ["dep:hf-hub"] # -> Enable downloading models from HuggingFace, relies on `tokio`
# ort forwarded
download-binaries = ["ort/download-binaries"]
copy-dylibs = ["ort/copy-dylibs"]
load-dynamic = ["ort/load-dynamic", "open_clip_inference/load-dynamic"]
# ort execution providers
cuda = ["ort/cuda"]
tensorrt = ["ort/tensorrt"]
nvrtx = ["ort/nvrtx"]
xnnpack = ["ort/xnnpack"]
webgpu = ["ort/webgpu"]
directml = ["ort/directml"]
coreml = ["ort/coreml"]
migraphx = ["ort/migraphx"]
openvino = ["ort/openvino"]
onednn = ["ort/onednn"]
qnn = ["ort/qnn"]
cann = ["ort/cann"]
nnapi = ["ort/nnapi"]
tvm = ["ort/tvm"]
acl = ["ort/acl"]
armnn = ["ort/armnn"]
vitis = ["ort/vitis"]
rknpu = ["ort/rknpu"]
azure = ["ort/azure"]

[[bench]]
name = "predict_benchmark"
harness = false