mirror of
https://github.com/storytold/lightcraft.git
synced 2026-10-26 19:17:02 -06:00
Andrew FrahnandClaude Opus 5.5
0242a75b5a
An accelerator hook in the face interpreter takes Conv nodes; lightcraft-gpu implements it with a WGSL implicit-GEMM convolution (any stride, padding, grouping; split-K for small maps; a direct kernel for depthwise layers), each model's weights resident on a device of its own. After loading, detection and recognition check the GPU in the background (agreement within 1e-3 and faster than the CPU) and otherwise stay on the CPU, which remains the reference; the first GPU error drops it. Sources: own design. Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
28 lines
1.2 KiB
Rust
28 lines
1.2 KiB
Rust
//! Face models for LightCraft: what a model is ([`ModelManifest`]), whether to trust a description of one
|
|
//! ([`manifest::validate`]), the models we know by hash ([`known`]), and a probe that reads an unknown
|
|
//! `.onnx` file's input and output shapes without running it ([`onnx`]), so installing a model of your
|
|
//! own can be one drop and one licence prompt.
|
|
//!
|
|
//! Everything here treats model files and manifests as hostile input: sizes are capped, numbers must be
|
|
//! finite and sane, and a malformed file is an error, never a panic. Detection and recognition run
|
|
//! on the same checked, single-thread pure-Rust CPU interpreter; an [`net::Accelerator`] (the GPU, in `lightcraft-gpu`) can
|
|
//! take its convolutions once it agrees with it.
|
|
#![forbid(unsafe_code)]
|
|
#![deny(clippy::unwrap_used, clippy::expect_used, clippy::panic, clippy::unimplemented, clippy::todo, clippy::unreachable)]
|
|
|
|
pub mod align;
|
|
pub mod catalog;
|
|
pub mod graph;
|
|
pub mod hash;
|
|
pub mod known;
|
|
pub mod manifest;
|
|
pub mod matching;
|
|
pub mod net;
|
|
pub mod onnx;
|
|
pub mod runtime;
|
|
pub mod suggest;
|
|
pub mod synthetic;
|
|
pub mod yunet;
|
|
|
|
pub use manifest::{Colour, Commercial, InputSpec, Licence, ManifestError, ModelManifest, OutputSpec, Resize, Role, Thresholds};
|