Python实例2-获取某文件夹下指定类型的图片名，完成图片格式的转换并保存

import random
import os
import os.path as osp
from PIL import Image

dataset_dir='/home/walden/pytorch/uiuc/data/VOC/cls'

def file_name(file_dir):
L=[]
for root, dirs, files in os.walk(file_dir):
for file in files:
if os.path.splitext(file)[1] == '.png':
#L.append(os.path.join(root, file))
L.append(file)
return L

my_filename=file_name(dataset_dir)

random.shuffle(my_filename)
f=open('all.txt','w')
for i in my_filename:
#k=' '.join([str(j) for j in i])
f.write(i+"\n")
f.close()

for i in my_filename:
path=osp.join(dataset_dir, '%s' % i)
img=Image.open(path)
img.save(path[0:len(path)-4]+'.bmp')