globox 0.1.0

Parse, convert and evaluate bounding box annotations
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

mod parse;
mod convert;
mod eval;

use parse::Parse;
use convert::Convert;
use eval::Eval;

use clap::{Parser, Subcommand, ValueEnum};

#[derive(Parser)]
#[command(version, about)]
struct Cli {
    #[arg(short, long, help = "Make the operations more talkative")]
    verbose: bool,

    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    /// Parse and display basic statistics
    Parse(Parse),

    /// Convert bounding box annotations between formats
    Convert(Convert),

    /// Evaluate bounding box annotations and predictions
    Eval(Eval),
}

#[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
enum SrcAnnFmt {
    Coco, Cvat, Imagenet, Labelme, OpenImage, Yolo, PascalVoc
}

#[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
enum DstAnnFmt {
    Coco, Cvat, Imagenet, Labelme, OpenImage, Yolo, PascalVoc, Vit
}

pub fn run() {
    let args = Cli::parse();

    match args.command {
        Commands::Parse(parse) => parse.run(),
        Commands::Convert(convert) => convert.run(),
        Commands::Eval(eval) => eval.run(),
    }
}