45 lines
1.1 KiB
TOML
45 lines
1.1 KiB
TOML
[package]
|
|
name = "rtx-image-classifier-demo"
|
|
version = "1.0.0"
|
|
edition = "2024"
|
|
rust-version = "1.92"
|
|
description = "Image classifier demo for RustyTorch++ - real-time CNN inference"
|
|
authors = ["RustyTorch Team"]
|
|
license = "MIT OR Apache-2.0"
|
|
repository = "https://github.com/rustytorch/rustytorch"
|
|
|
|
[dependencies]
|
|
image-classifier-shared = { path = "../image-classifier-shared" }
|
|
rtx-vision = { workspace = true }
|
|
rtx-tensor = { workspace = true }
|
|
rtx-backend = { workspace = true }
|
|
rtx-backend-cpu = { workspace = true }
|
|
|
|
# GPU backends (optional)
|
|
rtx-backend-cuda = { path = "../../crates/core/rtx-backend-cuda", optional = true }
|
|
rtx-backend-metal = { workspace = true, optional = true }
|
|
|
|
serde = { version = "1.0", features = ["derive"] }
|
|
serde_json = "1.0"
|
|
thiserror = "1.0"
|
|
tracing = "0.1"
|
|
base64 = "0.22"
|
|
image = { version = "0.25", default-features = false, features = ["png", "jpeg", "webp"] }
|
|
|
|
[features]
|
|
default = []
|
|
cuda = ["dep:rtx-backend-cuda"]
|
|
metal = ["dep:rtx-backend-metal"]
|
|
gpu = ["cuda", "metal"]
|
|
|
|
[dev-dependencies]
|
|
tempfile = "3.10"
|
|
|
|
[lints.rust]
|
|
unsafe_code = "forbid"
|
|
missing_docs = "warn"
|
|
|
|
[lints.clippy]
|
|
all = "warn"
|
|
pedantic = "warn"
|