Cocotools

cocotools 包提供了加载、操作/转换和可视化 COCO 格式数据集的工具。

设置

从 crates.io 获取该包。

API 使用

您可以在 此处 找到文档。

示例

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;

程序使用

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

未来功能

• 添加对关键点检测格式的支持。
• 添加从/到 PascalVOC 格式的转换。
• 添加从/到 SOLO 格式的转换。
• 在加载数据时验证数据，例如检查 sum(rle) == 图像中的像素数（在 crate 功能标志后面？）
• 在加载/转换数据时使用 rayon 吗？