latent-inspector 0.1.0

Fast CLI for inspecting and comparing learned representations across self-supervised vision models
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

mod cli;

// Re-export library modules so CLI subcommands can use `crate::` paths.
pub use latent_inspector::analysis;
pub use latent_inspector::dataset;
pub use latent_inspector::errors;
pub use latent_inspector::extract;
pub use latent_inspector::models;
pub use latent_inspector::tui;
pub use latent_inspector::validation;
pub use latent_inspector::viz;

use clap::Parser;
use tracing_subscriber::EnvFilter;

fn main() {
    let args = cli::Cli::parse();

    // Initialise tracing
    let filter = if args.verbose {
        "latent_inspector=debug,info"
    } else {
        "latent_inspector=info,warn"
    };

    tracing_subscriber::fmt()
        .with_env_filter(
            EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new(filter)),
        )
        .with_target(false)
        .with_writer(std::io::stderr)
        .without_time()
        .init();

    let result = match args.command {
        cli::Commands::Compare(a) => cli::compare::run(a),
        cli::Commands::Inspect(a) => cli::inspect::run(a),
        cli::Commands::Neighbors(a) => cli::neighbors::run(a),
        cli::Commands::Profile(a) => cli::profile::run(a),
        cli::Commands::Similarity(a) => cli::similarity::run(a),
        cli::Commands::Drift(a) => cli::drift::run(a),
        cli::Commands::Models(a) => cli::models::run(a),
        cli::Commands::Validate(a) => cli::validate::run(a),
        cli::Commands::Tui(a) => cli::tui::run(a),
    };

    if let Err(e) = result {
        if e.should_print() {
            eprintln!("error: {e}");
        }
        std::process::exit(e.exit_code());
    }
}