[package]
edition = "2024"
name = "vision-rs"
version = "0.1.2"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "A high-performance computer vision SDK for Rust."
homepage = "https://github.com/teenygrad/vision-rs"
documentation = "https://docs.teenygrad.org/api/vision-rs/vision_rs/"
readme = "README.md"
keywords = [
"computer-vision",
"yolo",
"detection",
"segmentation",
"tracking",
]
categories = [
"computer-vision",
"multimedia",
]
license = "Apache-2.0"
repository = "https://github.com/teenygrad/vision-rs"
[package.metadata.docs.rs]
no-default-features = true
[features]
cuda = [
"teeny-cuda",
"teeny-compiler",
"teeny-cli/cuda",
]
default = [
"cuda",
"training",
]
training = [
"teeny-kernels/training",
"teeny-core/training",
]
[lib]
name = "vision_rs"
path = "src/lib.rs"
[[example]]
name = "yolo26"
path = "examples/yolo26.rs"
[[test]]
name = "test_c2psa_backward_cuda"
path = "tests/test_c2psa_backward_cuda.rs"
[[test]]
name = "test_c2psa_cuda"
path = "tests/test_c2psa_cuda.rs"
[[test]]
name = "test_c2psa_yolo26"
path = "tests/test_c2psa_yolo26.rs"
[[test]]
name = "test_c3k2_backward_cuda"
path = "tests/test_c3k2_backward_cuda.rs"
[[test]]
name = "test_c3k2_cuda"
path = "tests/test_c3k2_cuda.rs"
[[test]]
name = "test_c3k2_variants_cuda"
path = "tests/test_c3k2_variants_cuda.rs"
[[test]]
name = "test_c3k2_yolo26"
path = "tests/test_c3k2_yolo26.rs"
[[test]]
name = "test_concat_upsample_cuda"
path = "tests/test_concat_upsample_cuda.rs"
[[test]]
name = "test_concat_upsample_yolo26"
path = "tests/test_concat_upsample_yolo26.rs"
[[test]]
name = "test_conv_grouped_backward_cuda"
path = "tests/test_conv_grouped_backward_cuda.rs"
[[test]]
name = "test_conv_grouped_cuda"
path = "tests/test_conv_grouped_cuda.rs"
[[test]]
name = "test_conv_yolo26"
path = "tests/test_conv_yolo26.rs"
[[test]]
name = "test_detect_cuda"
path = "tests/test_detect_cuda.rs"
[[test]]
name = "test_detect_decode"
path = "tests/test_detect_decode.rs"
[[test]]
name = "test_sigmoid_4d"
path = "tests/test_sigmoid_4d.rs"
[[test]]
name = "test_silu_4d"
path = "tests/test_silu_4d.rs"
[[test]]
name = "test_softmax_yolo"
path = "tests/test_softmax_yolo.rs"
[[test]]
name = "test_sppf_cuda"
path = "tests/test_sppf_cuda.rs"
[[test]]
name = "test_sppf_yolo26"
path = "tests/test_sppf_yolo26.rs"
[[test]]
name = "test_yolo26n_backward_cuda"
path = "tests/test_yolo26n_backward_cuda.rs"
[[test]]
name = "test_yolo26n_graph"
path = "tests/test_yolo26n_graph.rs"
[[test]]
name = "test_yolo_bce_cls_loss_cuda"
path = "tests/test_yolo_bce_cls_loss_cuda.rs"
[[test]]
name = "test_yolo_ciou_loss_cuda"
path = "tests/test_yolo_ciou_loss_cuda.rs"
[dependencies.anyhow]
version = "1.0"
default-features = false
[dependencies.teeny-cli]
version = "0.1.3"
optional = true
[dependencies.teeny-compiler]
version = "0.1.3"
optional = true
[dependencies.teeny-core]
version = "0.1.3"
features = ["std"]
[dependencies.teeny-cuda]
version = "0.1.3"
optional = true
[dependencies.teeny-data]
version = "0.1.3"
[dependencies.teeny-kernels]
version = "0.1.3"
[dependencies.teeny-macros]
version = "0.1.3"
[dependencies.teeny-triton]
version = "0.1.3"
[dev-dependencies.clap]
version = "4"
features = ["derive"]
[dev-dependencies.dotenv]
version = "0.15"
[dev-dependencies.eframe]
version = "0.29"
features = ["x11"]
[dev-dependencies.futures-util]
version = "0.3"
[dev-dependencies.image]
version = "0.25"
[dev-dependencies.indicatif]
version = "0.18"
[dev-dependencies.insta]
version = "1"
[dev-dependencies.reqwest]
version = "0.13"
features = ["stream"]
[dev-dependencies.serde]
version = "1"
features = ["derive"]
[dev-dependencies.serde_json]
version = "1"
[dev-dependencies.serial_test]
version = "3"
[dev-dependencies.teeny-compiler]
version = "0.1.3"
[dev-dependencies.tokio]
version = "1"
features = ["full"]
[dev-dependencies.toml]
version = "0.8"
[dev-dependencies.zip]
version = "2"