1.將圖片按順序重命名:
import os
path = "/home/customer/Desktop/voc/test2/"
filelist = os.listdir(path)
count=0
for file in filelist:
print(file)
for file in filelist: #遍歷所有文件
Olddir=os.path.join(path,file) #原來的文件路徑
if os.path.isdir(Olddir): #如果是文件夾則跳過
continue
filename=os.path.splitext(file)[0] #文件名
filetype=os.path.splitext(file)[1] #文件擴展名
Newdir=os.path.join(path,str(count).zfill(6)+filetype) #用字符串函數zfill 以0補全所需位數
os.rename(Olddir,Newdir) #重命名
count+=1
#命名改成000000.jpg格式。
2.將圖片名放進txt
import os
def generate(dir):
files = os.listdir(dir)
files.sort()
print ('****************')
print ('input :',dir )
print ('start...')
listText = open(dir+'/'+'test2.txt','w')
for file in files:
fileType = os.path.split(file)
if fileType[1] == '.txt':
continue
name = file + ' ' +'\n'
listText.write(name)
listText.close()
print ('down!')
print ('****************')
if __name__ == '__main__':
generate('/home/customer/Desktop/voc/test2')