mask 编码压缩

import pycocotools.mask as mask_util

def single_mask_to_rle(mask):
	rle = mask_util.encode(np.array(mask[:, :, None], order="F", dtype="uint8"))[0]
	rle["counts"] = rle["counts"].decode("utf-8")
	return rle

mask_rles = [single_mask_to_rle(mask) for mask in masks]

你可能感兴趣的:(python基础,python,深度学习,开发语言)