批量更改图像像素值，此处为更改mask的像素值

最新推荐文章于 2024-09-14 19:55:48 发布

Moksha262

最新推荐文章于 2024-09-14 19:55:48 发布

阅读量149

点赞数 2

文章标签： python

本文链接：https://blog.csdn.net/weixin_56314292/article/details/131562637

版权

import os
import cv2
from PIL import Image

img_path = "F:/CT_lung_seg_and_class/seg_data/CC-CCI_clean/train/mask/"
path_save = "F:/CT_lung_seg_and_class/seg_data/CC-CCI_clean/train/mask1/"
img_list = os.listdir(img_path)


for i in range(len(img_list)):
    img_name = img_list[i]
    img = cv2.imread(filename=img_path + "/" + img_name)
    img[img == 0] = 0
    img[img == 255] = 1
    img[img==165]=2
    img[img==80]=3

    cv2.imwrite(path_save+img_name.split(".")[0]+'.png', img)