mask rcnn图片掩码批量处理成uint8格式

最新推荐文章于 2021-03-14 19:26:22 发布

原创最新推荐文章于 2021-03-14 19:26:22 发布 · 770 阅读

2 ·

CC 4.0 BY-SA版权

Mask R-CNN 同时被 2 个专栏收录

2 篇文章

订阅专栏

mask 图片掩码处理

1 篇文章

订阅专栏

本篇博客介绍了一个Python脚本，该脚本利用PIL和numpy库批量读取指定目录下的图片及其对应的label.png文件，将这些label.png转换为具有特定色彩强度的图片并保存到新的目录下。此过程主要用于数据预处理，如图像分割任务中的掩膜转换。

from PIL import Image
import numpy as np
import os

j=0
path ='D:\Program Files\JetBrains\Mask_RCNN-master\mask_img'
save_path ='D:\Program Files\JetBrains\Mask_RCNN-master\mask_img\mask_img'
for img in os.listdir(path):
     for i in os.listdir(path+'/'+img):
         if i =="label.png":
             #print(i)
             # img =Image.fromarray(np.uint8(i))
             # img.save(path+img)
             img_ =Image.open(path+'/'+img+'/'+ i)
             print(img_)
             img_ = Image.fromarray(np.uint8(img_)*20)
             #print(type(img_))
             img_.save(save_path+'/'+str(j)+'.png')
             j=j+1