详解COCO数据格式的json文件内容

coco标注格式的json文件详解

# coco_json结构示意
{
    'info':info,
    'licenses':[licenses],
    'images':[image],
    'annotations':[annotation]
}

1. **`images`**:这个部分包含了所有图像的信息。每个图像都表示为一个字典,包含以下字段:
   - `id`:唯一标识图像的ID。
   - `file_name`:图像文件的文件名。
   - `width`:图像宽度(以像素为单位)。
   - `height`:图像高度(以像素为单位)。
   - `license`:图像的许可证信息(可选)。

2. **`annotations`**:这个部分包含了与图像中对象实例分割相关的注释信息。每个注释表示为一个字典,包含以下字段:
   - `id`:唯一标识注释的ID。
   - `image_id`:与注释相关联的图像的ID。
   - `category_id`:对象的类别ID,对应于`categories`部分中的类别。
   - `segmentation`:对象的分割掩码。通常表示为多边形或掩码的像素坐标。
   - `area`:对象的像素面积。
   - `bbox`:对象的边界框,格式为`[x, y, width, height]`。
   - `iscrowd`:标志位,指示对象是否是“杂乱”(例如,一群对象被视为单个对象)。

3. **`categories`**:这个部分包含了对象类别的信息。每个类别表示为一个字典,包含以下字段:
   - `id`:唯一标识类别的ID。
   - `name`:类别的名称。
   - `supercategory`:类别的超类别,用于组织相关类别。

4. **`info`**:这个部分包含了关于数据集的一般信息,如数据集名称、描述、版本等。

5. **`licenses`**:这个部分包含了与数据集许可相关的信息,如许可证名称、ID、URL等。

以下是一个简化的coco格式json示例:

{
  "images": [
    {
      "id": 1,
      "file_name": "image1.jpg",
      "width": 640,
      "height": 480
    },
    {
      "id": 2,
      "file_name": "image2.jpg",
      "width": 800,
      "height": 600
    }
  ],
  "annotations": [
    {
      "id": 1,
      "image_id": 1,
      "category_id": 1,
      "segmentation": [[x1, y1, x2, y2, ...]],
      "area": 1234,
      "bbox": [x, y, width, height],
      "iscrowd": 0
    },
    {
      "id": 2,
      "image_id": 1,
      "category_id": 2,
      "segmentation": [[x1, y1, x2, y2, ...]],
      "area": 567,
      "bbox": [x, y, width, height],
      "iscrowd": 0
    }
  ],
  "categories": [
    {
      "id": 1,
      "name": "person",
      "supercategory": "human"
    },
    {
      "id": 2,
      "name": "car",
      "supercategory": "vehicle"
    }
  ],
  "info": {
    "description": "COCO 2017 dataset",
    "version": "1.0",
    "year": 2017,
    "contributor": "Microsoft COCO group",
    "url": "http://cocodataset.org"
  },
  "licenses": [
    {
      "id": 1,
      "name": "CC BY-SA 2.0",
      "url": "https://creativecommons.org/licenses/by-sa/2.0/"
    }
  ]
}

在COCO数据集的`annotations`字段中,`segmentation`字段用于描述对象实例的分割信息。`segmentation`字段的内容可以是多边形(polygon)或二进制掩码(mask),具体格式取决于数据集的标注方式。以下是关于`segmentation`字段的详细介绍:

1. **多边形表示(Polygon Representation)**:

   在COCO数据集中,`segmentation`字段通常以多边形的形式来表示对象实例的分割区域。多边形表示是一个列表,其中包含一系列坐标点,这些点按照顺序连接以形成多边形边界。坐标点的顺序是按照顺时针或逆时针方向排列的。

   例如,`segmentation`字段的内容可以如下所示:
   ```
   "segmentation": [[x1, y1, x2, y2, x3, y3, ...]]
   ```

   其中,每对(x, y)表示一个多边形边界上的点坐标。这些坐标点按照顺时针或逆时针的顺序排列。

2. **二进制掩码表示(Binary Mask Representation)**:

   在某些情况下,COCO数据集也可以使用二进制掩码来表示对象实例的分割区域。二进制掩码是一个二维矩阵,其中每个像素都标识为前景(对象)或背景。通常,前景像素用1表示,背景像素用0表示。

   例如,`segmentation`字段的内容可以如下所示:
   ```
   "segmentation": {
       "size": [height, width],
       "counts": "binary_mask_encoded"
   }
   ```

   其中,`size`字段包含掩码的高度和宽度,`counts`字段包含了用一种编码方式表示的二进制掩码。

在实例分割的coco-json中,annotations--segmentation以counts形式存储分割信息,且用到了RLE编码,因此利用segmentation存储的分割信息还需要进行RLE解码操作,这里可以利用pycocotools中的方法进行解码:

import pycocotools.mask as mask_utils
from pycocotools.coco import COCO

# 读取json
coco = COCO(json_path)
images_ids = coco.getImgIds()

# 逐图像处理
for img_id in images_ids:
    img_info = coco.loadImgs(img_id)[0]
    ann_ids = coco.getAnnIds(imgIds=img_id)
    anns = coco.loadAnns(ann_ids)

    # 逐实例处理
    for ann in anns:
        rle = coco.annToRLE(ann)        # 解码
        mask = mask_utils.decode(rle)   # 生成原图mask
        
        mask[mask == 1] = 255           # 调整灰度值便于显示
       

你可能感兴趣的:(Working,人工智能,算法)