coco mask处理

目录

coco可视化：

---

coco可视化：

import pycocotools
from pycocotools.coco import COCO
import numpy as np
if __name__ == '__main__':

    DATA_ROOT=r"C:\Users\Administrator\Documents\工具\output\output\video_1"
    coco_filepath = r"C:\Users\Administrator\Documents\工具\output\output\video_1\person_keypoints.json"
    coco = COCO(coco_filepath)

    impath_to_examples = {}
    for ann in coco.anns.values():
        filename = coco.imgs[ann['image_id']]['file_name']
        image_path = f'{DATA_ROOT}/images/{filename}'

        joints = np.array(ann['keypoints']).reshape([-1, 3])
        visibilities = joints[:, 2]
        coords = joints[:, :2].astype(np.float32).copy()

import pycocotools._mask as _mask

frPyObjects = _mask.frPyObjects


segm = ann['segmentation']
if type(segm) == list:
    # polygon -- a single object might consist of multiple parts
    # we merge all parts into one mask rle code
    rles = maskUtils.frPyObjects(segm, h, w)
    rle = maskUtils.merge(rles)
elif type(segm['counts']) == list:
    # uncompressed RLE
    rle = maskUtils.frPyObjects(segm, h, w)
else:
    # rle
    rle = ann['segmentation']
return rle