python 8位单通道jpg转8位单通道png脚本

目的

PaddleSeg的输入，需要png格式的单通道8位图片输入。将某数据集的8位单通道jpg mask转换为PaddleSeg的输入标准。

Code

import numpy as np
import os
import PIL.Image as Image
path = './data/ct/Annotations'
save_path = './data/ct/Anno_png'

for root, dirs, files in os.walk(path):
        for file in files:
            im = Image.open(os.path.join(root, file))
            lbl = np.asarray(im)
            lbl_pil = Image.fromarray(lbl.astype(np.uint8), mode='L')
            lbl_pil.save(os.path.join(save_path, file.replace('.jpg', '.png')))