python下的图片转换

因为:

做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')
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。