mirror of
https://github.com/storytold/lightcraft.git
synced 2026-10-26 11:09:23 -06:00
akvaithiandClaude Opus 5.5
84b8995d7b
- `lightcraft_denoise::remote`: protocol v1 over TCP (JSON lines + byte-shuffled, deflated half-float tiles), a `Remote` client (connection pool, four slots, greeting checks the model's SHA-256 and tile size, gives up after three failed tiles) and `serve` (shared token compared in constant time, size caps including payloads that inflate past their tile, connection limit, timeouts). - Engine: with LIGHTCRAFT_DENOISE_URL / LIGHTCRAFT_DENOISE_TOKEN set, the loaded model greets the server on first use and adds its slots to this computer's tiles at once; a tile goes to a free server slot, else runs here, and a tile the server fails runs here too. `denoise.status` → `device.remote` says where. - `lightcraft-cli denoise-serve` serves the installed, selected model on this computer's card (checked against the CPU, as in the app) or CPU. - docs/denoise.md: setup, safety (unencrypted: VPN/LAN only) and the protocol. Tested: protocol round trips, refusals (token, model, tile size, hostile payloads), fallback and give-up; the engine model with a live, a mismatched and a dead server; the CLI binary serving a synthetic model end to end. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
40 lines
2.0 KiB
Rust
40 lines
2.0 KiB
Rust
//! AI denoise for LightCraft, as a step between the sensor and the demosaic that the rest of the develop
|
|
//! pipeline never sees: the raw file stays untouched, the denoised picture is cached data that can always be
|
|
//! made again, and the amount is an ordinary develop setting.
|
|
//!
|
|
//! - [`manifest`]: what a denoise model takes and gives, and who may use it (untrusted JSON, validated).
|
|
//! - [`known`] and [`archive`]: the models LightCraft can fetch, and taking the model file out of the `.zip` it comes in.
|
|
//! - [`bayer`]: Bayer layouts and the packing of a mosaic into the four planes a model takes.
|
|
//! - [`tiles`] and [`run`]: cutting a picture into overlapping tiles for a fixed-size model, bringing each tile's
|
|
//! output scale back to the input's, keeping clipped highlights as they were, and blending the tiles together.
|
|
//! The model itself is a [`run::TileRunner`], so all of this is tested without one.
|
|
//! - [`product`]: the cache file the denoised picture is kept in (half floats, compressed in strips) and the
|
|
//! blend with the plain demosaic that the Amount slider controls.
|
|
//! - [`cpu`] and [`runtime`]: a single-thread pure-Rust CPU runner and its installation self-test.
|
|
//! - [`net`] and [`onnx`]: checked plain network data and bounded ONNX decoding, shared by CPU and GPU.
|
|
//! - [`remote`]: running tiles on another computer's graphics card over TCP (a [`run::TileRunner`] and its server).
|
|
//!
|
|
//! Everything here treats model files, manifests and cache files as hostile input: sizes are capped, numbers must
|
|
//! be finite, and a malformed file is an error, never a panic.
|
|
#![forbid(unsafe_code)]
|
|
#![deny(clippy::unwrap_used, clippy::expect_used, clippy::panic, clippy::unimplemented, clippy::todo, clippy::unreachable)]
|
|
|
|
#[cfg(feature = "runtime")]
|
|
pub mod cpu;
|
|
|
|
pub mod net;
|
|
pub mod onnx;
|
|
mod onnx_proto;
|
|
|
|
pub mod reference;
|
|
pub mod remote;
|
|
|
|
#[cfg(feature = "runtime")]
|
|
pub mod runtime;
|
|
#[doc(hidden)]
|
|
pub mod synthetic;
|
|
|
|
pub use lightcraft_denoise_core::*;
|
|
|
|
pub mod synthetic_proto;
|