bmp图像转jpg文件

评论(Comments): 1
阅读量(Views):335

预计阅读时间：1 分钟

针对大批量的图像文件需要进行格式转化, 比如将 bmp 转化为jpg.

from PIL import Image
import numpy as np
import argparse
import os

parser = argparse.ArgumentParser(description='bmp2jpg script')
parser.add_argument("--read_dir", type=str)
parser.add_argument("--write_dir", type=str)
args = parser.parse_args()

def trans_format():
    img_file = os.listdir(args.read_dir)
    for name in img_file:
        img = Image.open(args.read_dir + name)
        #img = img.convert("RGB")
        #img = np.array(img)[...,::-1]
        #img = Image.fromarray(img)
        name = name.split('.')
        if str(name[-1]).startswith("png"):
            name[-1] = "jpg"
            name = ".".join(name)
            to_save_path = args.write_dir + name
            img.save(to_save_path)


if __name__ == "__main__":
    trans_format()

调用示例

python bmp2jpg.py --read_dir "/dataset/" --write_dir "/dataset/"

可以通过 tree -l 命令快速查看格式转化后的图像.

本文由 admin 原创，转载请注明出处。若存在侵权请联系删除。