python bmp转jpg 且灰度图转彩色

近期因为研究tensorflow,需要bmp位图转jpg格式,并且原图为灰度图,要转为RGB图像,在网上搜,发现灰度转RGB的python文章几乎没有,其实就一句代码,很简单,现贴出原代码。

# coding:utf-8
import os
from PIL import Image

# bmp 转换为jpg,灰度图转RGB
def bmpToJpg_grayToRGB(file_path):
   for fileName in os.listdir(file_path):
       print(fileName)
       newFileName = fileName[0:fileName.find(".bmp")]+".jpg"
       print(newFileName)
       im = Image.open(file_path+"\\"+fileName)
       rgb = im.convert('RGB')      #灰度转RGB
       rgb.save(file_path+"\\"+newFileName)

# 删除原来的位图
def deleteImages(file_path, imageFormat):
   command = "del "+file_path+"\\*."+imageFormat
   os.system(command)

def main():
   file_path = "D:\\models-master\\research\\object_detection\\images"
   bmpToJpg_grayToRGB(file_path)
   deleteImages(file_path, "bmp")

if __name__ == '__main__':
   main()

文件路径可自行修改。

你可能感兴趣的:(python bmp转jpg 且灰度图转彩色)