批量修改文件后缀.py

import os

sonDirName="data/train"

filelist = os.listdir(sonDirName)
total_num = len(filelist)
i = 1
for item in filelist:
    # print(item)
    if item.endswith('.jpeg') or item.endswith('.png') or item.endswith('.bmp'):
    # if item.endswith('.xml'):
        src = os.path.join(os.path.abspath(sonDirName), item)
        dst = os.path.join(os.path.abspath(sonDirName), item.split(".")[0]+'.jpg')
        # print(dst)
        try:
            os.rename(src, dst)
            print('converting %s to %s ...' % (src, dst))
            i = i + 1
        except:
            continue

print('total %d to rename & converted %d jpgs' % (total_num, i - 1))