Coco数据集中的rle格式处理

1:rle转mask

如果是rle的格式,则其是一个字典,包含size和counts两个键,形如:

rle = {'size':[512,512],'counts': '一个很长的字符串乱码'}

 此时该如何转化为可以训练的mask呢?

很简单,import pycocotools.mask as mask_utils:

 直接利用下面这条命令就可以得到二值mask

a = np.array(mask_util.decode(rle), dtype=np.float32)


 2:mask转rle

参考自:Segmentaion标签的三种表示:poly、mask、rle_Adenialzz的博客-CSDN博客


————————————————
原文链接:https://blog.csdn.net/xx_xjm/article/details/126608359

猜你喜欢

转载自blog.csdn.net/dou3516/article/details/132028394