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博客

 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐