burn_dragon_vision 0.4.0

Foveation and vision sampling utilities for burn dragon
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
#![recursion_limit = "256"]

pub mod foveation;
#[cfg(feature = "train")]
pub mod train;
pub mod wgsl;

pub use foveation::{
    CpuImageLevel, CpuPyramidCache, FoveaWarpMode, PyramidMode, build_pyramid_cache,
    image_from_nchw, lod_sigma_from_sigma, render_foveated_patch,
    render_foveated_patch_with_radius, sigma_from_unit,
};

pub use wgsl::{FOVEATION_BUFFER_SHADER, FOVEATION_SHADER, PYRAMID_SHADER, SCATTER_BUFFER_SHADER};