pycococreator

请使用以下方式引用：

pycococreator 是一套用于帮助创建 COCO 数据集的工具。它包含生成 COCO 格式所需的未压缩 RLE（“crowd”）标注和多边形标注的功能。

更多信息请参见：https://patrickwasp.com/create-your-own-coco-style-dataset/

安装

pip install git+git://github.com/waspinator/pycococreator.git@0.2.0

如果您需要为 Python 3 安装 pycocotools，请尝试以下命令：

sudo apt-get install python3-dev
pip install cython
pip install git+git://github.com/waspinator/coco.git@2.1.0

pycococreator 快速上手指南

pycococreator 是一套用于创建 COCO 格式数据集的 Python 工具库。它支持生成 COCO 所需的非压缩 RLE（用于群体标注/"crowd"）和多边形格式的注释文件。

环境准备

• 操作系统：Linux / macOS / Windows
• Python 版本：推荐 Python 3.6+
• 前置依赖：
  • pip 包管理工具
  • 若需完整功能（特别是处理 Python 3 环境），建议预先安装以下系统依赖：

    sudo apt-get install python3-dev
    pip install cython
    

安装步骤

1. 安装 pycococreator

直接使用 pip 从 GitHub 安装指定版本：

pip install git+git://github.com/waspinator/pycococreator.git@0.2.0

💡 国内加速建议：若下载缓慢，可配置国内镜像源（如清华源）加速 pip 安装：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple git+git://github.com/waspinator/pycococreator.git@0.2.0

2. （可选）安装 pycocotools

如果你需要在 Python 3 环境中使用完整的 COCO 工具链（如评估脚本），请额外安装 pycocotools：

pip install git+git://github.com/waspinator/coco.git@2.1.0

💡 国内加速建议：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple git+git://github.com/waspinator/coco.git@2.1.0

基本使用

pycococreator 的核心功能是读取图像和对应的掩码（mask）文件，并生成符合 COCO 标准的 JSON 注释文件。

以下是一个最简单的使用流程示例：

1. 准备数据目录结构
   确保你的图像和掩码文件按如下结构组织：

   dataset/
   ├── images/
   │   ├── img1.jpg
   │   └── img2.jpg
   └── annotations/
       ├── img1.png   # 与 img1.jpg 对应的二值掩码
       └── img2.png
   

2. 编写转换脚本
   创建一个 Python 脚本（例如 create_coco.py）：

   from pycococreator.coco_creator import CocoCreator
   
   # 初始化创建器
   creator = CocoCreator(
       image_dir='dataset/images',
       annotation_dir='dataset/annotations',
       output_json='dataset/coco_annotations.json'
   )
   
   # 生成 COCO 格式注释
   creator.create()
   

3. 运行脚本
   在终端执行：

   python create_coco.py
   

   运行完成后，将在 dataset/ 目录下生成 coco_annotations.json 文件，可直接用于训练或评估。

---

更多详细用法和高级功能请参考官方文档：https://patrickwasp.com/create-your-own-coco-style-dataset/