1. 程式人生 > 程式設計 >python實現遍歷資料夾圖片並重命名

python實現遍歷資料夾圖片並重命名

在做深度學習相關專案時,需要標註圖片,篩選過後圖片名字帶有括號,顯得比較亂,因此利用python進行統一規範重新命名操作

python實現遍歷資料夾圖片並重命名

實現方法是利用python的os模組對資料夾進行遍歷(listdir),然後使用rename進行改名操作

程式碼如下

# -*- coding:utf8 -*-
 
import os
 
class BatchRename():
 '''
 批量重新命名資料夾中的圖片檔案
 '''
 def __init__(self):
 self.path = 'C:/Users/lenovo/Desktop/label5/3'
 
 def rename(self):
 filelist = os.listdir(self.path)
 total_num = len(filelist)
 i = 1
 for item in filelist:
 if item.endswith('.jpg'):
 src = os.path.join(os.path.abspath(self.path),item)
 dst = os.path.join(os.path.abspath(self.path),'02'+ str(i) + '.jpg')
 try:
  os.rename(src,dst)
  print 'converting %s to %s ...' % (src,dst)
  group = 0
 except:
  continue
 if item.endswith('.xml'):
 src = os.path.join(os.path.abspath(self.path),'02'+ str(i) + '.xml')
 try:
  os.rename(src,dst)
  group = 1
  
 except:
  continue
 if group:
 i = i + 1
 print 'total %d to rename & converted %d jpgs' % (total_num,i)
 
if __name__ == '__main__':
 demo = BatchRename()
 demo.rename()

執行後效果如下

python實現遍歷資料夾圖片並重命名

簡潔版(tab補全)

# -*- coding:utf8 -*-
import os
path = 'SegmentationClassRaw/'
filelist = os.listdir(path)
for item in filelist:
 #print('item name is ',item)
 if item.endswith('.png'):
 name = item.split('.',3)[0] + '.' + item.split('.',3)[1]
 src = os.path.join(os.path.abspath(path),item)
 dst = os.path.join(os.path.abspath(path),name + '.png')
 try:
 os.rename(src,dst)
 print('rename from %s to %s'%(src,dst))
 except:
 continue

參考:利用Python對資料夾下圖片資料進行批量改名的程式碼例項

本文已被收錄到專題《python圖片處理操作》 ,歡迎大家點選學習更多精彩內容。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。