Python开发---语义分割标注转图片掩膜

from PIL import Image,ImageDraw
import json
import os
def json2jpg(annosFile,imgsDir,masksDir,colorMap,isAlpha,backgroundColor):
    f=open(annosFile,'r')
    annos=json.load(f)
    f.close()
    for file,anno in annos.items():
        print(file)
        #读取原始图像大小
        rawFile=os.path.join(imgsDir,anno['filename'])
        
        rawImg=Image.open(rawFile)
        #新建同样大小画布,jpg格式还是png格式
        mode='RGBA' if isAlpha else 'RGB'
        ext='.png' if isAlpha else '.jpg'
        maskImg=Image.new(mode,rawImg.size,backgroundColor)
        draw=ImageDraw.Draw(maskImg)
        for index,region in anno['regions'].items():
            print(index)
            x=region['shape_attributes']['all_points_x']
            y=region['shape_attributes']['all_points_y']
            label=region['region_attributes']['label']
            x=map(int,x)
            y=map(int,y)
            #使用标注类别对应的颜色标注每一个标注区域
            draw.polygon(list(zip(x,y)),fill=colorsMap.get(label,backgroundColor))
        maskFile=os.path.join(masksDir,os.path.splitext(anno['filename'])[0])
        maskImg.save(maskFile+ext)



if __name__=='__main__':
    #定义各个标注类别对应的颜色
    colorsMap={'foreground':'#00ff00','body':'yellow'}
    json2jpg('labels.json','pics/','masks/',colorsMap,True,'black')