赞
踩
本人编程萌新,从机械电子专业半路出家。在修改数据集通道数时,发现搜集到的结果无法正常运转。故将编写结果分享给同样需要的朋友。
程序亲测,可运行。
- 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]

本人萌新,初次发稿。若有问题,欢迎指正。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。