Skip to content

Latest commit

 

History

History

cocotools

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 

Cocotools

cocotools ci Crate Minimum rustc 1.64 Documentation

The cocotools crate provides tools to load, manipulate/convert and visualize COCO format datasets.

Setup

Get the crate from crates.io.

API Usage

You can find the documentation here.

Example

use std::path::PathBuf;
use cocotools::COCO;

let annotations_file_path = PathBuf::from("../data_samples/coco_25k/annotations.json");
let image_folder_path = PathBuf::from("../data_samples/coco_25k/images");
let coco_dataset = COCO::new(&annotations_file_path, &image_folder_path)?;
let file_name = dataset.get_img(17627)?.file_name;

Program Usage

cargo run -- visualize  ../data_samples/coco_25k/annotations.json ../data_samples/coco_25k/images -s 000000017627
cargo run -- convert-segmentation ../data_samples/coco_25k/annotations.json rle -o annotations_rle.json

Future features

  • Add support for keypoint detection format.
  • Add conversion from/to PascalVOC format.
  • Add conversion from/to SOLO format.
  • Add validation of the data when loading it, for example check that sum(rle) == nb pixels in the image (behind a crate feature flags ?)
  • Use rayon when loading/converting the data ?