cocotools 0.0.7

Package providing functionalities to work with COCO format datasets.
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

use std::error;
use std::path::PathBuf;

use clap::Parser;

mod argparse;
mod coco;
mod errors;
mod mask;
mod utils;
mod visualize;
use crate::argparse::{Cli, Commands};
use crate::coco::COCO;
use crate::visualize::display;

fn main() -> Result<(), Box<dyn error::Error>> {
    let cli = Cli::parse();

    match &cli.command {
        Commands::Visualize {
            annotations_file,
            image_folder,
            sample_id,
        } => {
            let dataset = COCO::new(annotations_file, image_folder)?;
            if let Some(sample_id) = sample_id {
                let img = dataset.draw_img_anns(*sample_id, true)?;
                display::img(&img, &dataset.get_img(*sample_id)?.file_name)?;
            } else {
                for img_entry in dataset.get_imgs() {
                    let img = dataset.draw_img_anns(img_entry.id, true)?;
                    display::img(&img, &img_entry.file_name)?;
                }
            }
        }
        Commands::ConvertSegmentation {
            annotations_path,
            target_segmentation,
            output_path,
        } => {
            let mut dataset = COCO::new(annotations_path, &PathBuf::from("N/A"))?;
            mask::conversions::convert_coco_segmentation(&mut dataset, *target_segmentation)?;
            let output_path = output_path
                .as_ref()
                .map_or_else(|| annotations_path, |output_path| output_path);
            dataset.save_to(output_path)?;
        }
    }
    Ok(())
}