美文网首页
python下的图片转换

python下的图片转换

作者: 危险的小伙伴 | 来源:发表于2019-07-29 12:40 被阅读0次

因为:

做yolov3-tiny的项目,我的训练数据都是BMP格式的

所以:

import os 
from PIL import Image 
import shutil 
import sys 

output_dirHR = 'jpg_out'

if not os.path.exists(output_dirHR):
    os.makedirs(output_dirHR)

def img2img(dataset_dir,type): 
    files = [] 
    image_list = os.listdir(dataset_dir) 
    files = [os.path.join(dataset_dir, _) for _ in image_list] 
    for index,bmp in enumerate(files):
        if index > 100000: 
            break
        try: 
            sys.stdout.write('\r>>Converting image %d/100000 ' % (index)) 
            sys.stdout.flush() 
            im = Image.open(bmp) 
            img = os.path.splitext(bmp)[0] + "." + type
            im.save(img) 
            shutil.move(img, output_dirHR) 
        except IOError as e: 
            print('could not read:',bmp) 
            print('error:',e) 
            print('skip it\n') 
    sys.stdout.write('Convert Over!\n') 
    sys.stdout.flush()

if __name__ == "__main__": 
    current_dir = os.getcwd() 
    print(current_dir)
    img2img(current_dir,'jpg')

相关文章

网友评论

      本文标题:python下的图片转换

      本文链接:https://www.haomeiwen.com/subject/dufbrctx.html