python批量处理语义分割图片代码记录

最新推荐文章于 2023-10-25 12:04:49 发布

qq_45932469

最新推荐文章于 2023-10-25 12:04:49 发布

阅读量511

点赞数

分类专栏：语义分割代码复现文章标签： python numpy opencv

本文链接：https://blog.csdn.net/qq_45932469/article/details/127867290

版权

语义分割代码复现专栏收录该内容

2 篇文章 0 订阅

订阅专栏

import cv2
import os
import numpy as np
from PIL import Image
import skimage.io as io

def colorful(img):
    '''
    img:需要上色的图片
    
    '''
    img = Image.fromarray(img)  # 将图像从numpy的数据格式转为PIL中的图像格式
    palette = []
    for i in range(256):
        palette.extend((i, i, i))
    palette[:3 * 21] = np.array([[0, 0, 0],
                             [128, 0, 0],
                             [0, 128, 0],
                             [128, 128, 0],
                             [0, 0, 128],
                             [128, 0, 128],
                             [0, 128, 128],
                             [128, 128, 128],
                             [64, 0, 0],
                             [192, 0, 0],
                             [64, 128, 0],
                             [192, 128, 0],
                             [64, 0, 128],
                             [192, 0, 128],
                             [64, 128, 128],
                             [192, 128, 128],
                             [0, 64, 0],
                             [128, 64, 0],
                             [0, 192, 0],
                             [128, 192, 0],
                             [0, 64, 128]
                             ], dtype='uint8').flatten()

    img.putpalette(palette)
    return img

data_dir = "使用你自己的要处理的文件夹绝对路径"
for filename in os.listdir(data_dir):
    print(filename)
    img = cv2.imread(data_dir+'/'+filename,cv2.IMREAD_GRAYSCALE)
    img = colorful(img)
    img.save('存储路径'+"/"+filename)