1. 程式人生 > >【python】儲存某個資料夾下所有圖片名字到一個txt檔案裡

【python】儲存某個資料夾下所有圖片名字到一個txt檔案裡

用python實現讀取某一資料夾下的所有圖片名字到一個txt檔案

import os
dir1='/data/Datasets/JPEGImages'#圖片檔案存放地址
txt1 = '/data/Datasets/picture.txt'#圖片檔名存放txt檔案地址
f1 = open(txt1,'a')#開啟檔案流
for filename in os.listdir(dir1):
    f1.write(filename.rstrip('.jpg'))#只儲存名字,去除字尾.jpg
    f1.write("\n")#換行
f1.close()#關閉檔案流