近期因為研究tensorflow,需要bmp位圖轉jpg格式,并且原圖為灰度圖,要轉為RGB圖像,在網上搜,發(fā)現灰度轉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()
文件路徑可自行修改。