当前位置:   article > 正文

将指定目录下的所有黑白图片转为彩色格式_python 怎么让黑白照片变彩色

python 怎么让黑白照片变彩色

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

程序亲测,可运行。

  1. import os
  2. from PIL import Image
  3. from tqdm import tqdm
  4. import numpy as np
  5. img_path = './images_test1000_01' #填入图片所在文件夹的路径
  6. img_Topath = './images_test1000_01_RGB' #填入图片转换后的文件夹路径
  7. dirs = os.listdir(img_path)
  8. for item in tqdm(range(len(dirs)), desc='Processing'): # 导入进度条库tqdm
  9. img_name = dirs[item]
  10. image_path=os.path.join(img_path,img_name) # os.path.join拼接 地址img_path、图片名img_name
  11. # ./images_test1000_01\.
  12. img=Image.open(image_path)
  13. if(img.mode!='RGB'):
  14. img = img.convert("RGB")
  15. # img = np.array(img) # 转为array查看名称与格式
  16. # print(img_name)
  17. # print(img.shape)
  18. # img = Image.fromarray(img) # 转为array后必须用Image.fromarray转回image格式,
  19. img.save(img_Topath +'/'+img_name) # 否则会报错:'numpy.ndarray' object has no attribute 'save'
  20. # result:Processing: 100%|██████████| 1000/1000 [00:02<00:00, 378.22it/s]

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/煮酒与君饮/article/detail/983068?site
推荐阅读
相关标签
  

闽ICP备14008679号