将指定目录下的所有黑白图片转为彩色格式

本人编程萌新,从机械电子专业半路出家。在修改数据集通道数时,发现搜集到的结果无法正常运转。故将编写结果分享给同样需要的朋友。

程序亲测,可运行。

import os
from PIL import Image
from tqdm import tqdm
import numpy as np
 
img_path = './images_test1000_01'        #填入图片所在文件夹的路径
img_Topath = './images_test1000_01_RGB'       #填入图片转换后的文件夹路径
dirs = os.listdir(img_path)

for item in tqdm(range(len(dirs)), desc='Processing'):     # 导入进度条库tqdm
    img_name = dirs[item]
    image_path=os.path.join(img_path,img_name)   # os.path.join拼接 地址img_path、图片名img_name
    # ./images_test1000_01\.
    img=Image.open(image_path)
    if(img.mode!='RGB'):   
        img = img.convert("RGB")
        # img = np.array(img)       # 转为array查看名称与格式
        # print(img_name)
        # print(img.shape)
        # img = Image.fromarray(img)        # 转为array后必须用Image.fromarray转回image格式,
    img.save(img_Topath +'/'+img_name)      # 否则会报错:'numpy.ndarray' object has no attribute 'save'

# result:Processing: 100%|██████████| 1000/1000 [00:02<00:00, 378.22it/s]

本人萌新,初次发稿。若有问题,欢迎指正。


版权声明:本文为m0_65567747原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
THE END
< <上一篇
下一篇>>