将指定目录下的所有黑白图片转为彩色格式
本人编程萌新,从机械电子专业半路出家。在修改数据集通道数时,发现搜集到的结果无法正常运转。故将编写结果分享给同样需要的朋友。
程序亲测,可运行。
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 版权协议,转载请附上原文出处链接和本声明。