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
参考自:(3条消息) Segmentaion标签的三种表示:poly、mask、rle_Adenialzz的博客-CSDN博客

更多推荐

所有评论(0)