將PPM格式圖片轉(zhuǎn)化為JPG格式

做圖像識(shí)別的時(shí)候數(shù)據(jù)集常常是ppm格式的，雖然不影響建模訓(xùn)練，但是我們電腦往往不支持ppm格式的圖像展示。
比如到做交通標(biāo)志識(shí)別的時(shí)候用到的BelgiumTS交通數(shù)據(jù)集或者德國(guó)GTSRB數(shù)據(jù)集
下載后得到都是ppm格式的圖像。

格式轉(zhuǎn)化代碼

from PIL import Image import osinput_train_path = r"E:\test\BelgiumTSC_Training\Training" output_train_path = r"E:\BelgiumTSC_JPG\Train_data" input_test_path = r"E:\test\BelgiumTSC_Testing\Testing" output_test_path = r"E:\BelgiumTSC_JPG\Test_data"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 -1directories = [d for d in os.listdir(in_dir) if os.path.isdir(os.path.join(in_dir, d))]for d in directories:# 每一類的路徑label_directory = os.path.join(in_dir, d)new_directory = os.path.join(out_dir, d)out_folder = os.path.exists(out_dir + d)if not out_folder:os.mkdir(new_directory)file_names = [os.path.join(label_directory, f) for f in os.listdir(label_directory) if f.endswith(".ppm")]# file_names is every photo which is end with ".ppm"count = 0for files in file_names:file_path, extfilename = os.path.split(files)filename, extname = os.path.splitext(extfilename)out_file = filename + '.jpg'# print(filepath,',',filename, ',', out_file)im = Image.open(files)new_path = os.path.join(new_directory, out_file)print(count, ',', new_path)count = count + 1im.save(new_path)if __name__ == '__main__':batch_image(input_test_path, output_test_path)batch_image(input_train_path, output_train_path)

轉(zhuǎn)化后會(huì)保留原有的兩級(jí)目錄，方便建模訓(xùn)練。

總結(jié)

以上是生活随笔為你收集整理的批量将PPM格式图片转化为JPG格式的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。