使用keras进行行人检测分类器的时候,使用了Daimler Pedestrian Classification Benchmark Dataset数据集,由于图像为 .pgm 格式,而 Keras 的图片生成器 ImageDataGenerator 仅支持 JPG、PNG和BNP。
本来打算写 python 脚本使用 PIL 将其转换成 .jpg 格式的。
1.python pgm转jpg的脚本代码
from PIL import Image
import os, glob
def batch_image(in_dir, out_dir):
if not os.path.exists(out_dir):
print(out_dir, 'is not existed.')
os.mkdir(out_dir)
if not os.path.exists(in_dir):
print(in_dir, 'is not existed.')
return -1
count = 0
for files in glob.glob(in_dir+'/*'):
filepath, filename = os.path.split(files)