圖像數據集之python批量重命名

在訓練模型之前要對數據集進行規範化,此文提供了簡便易行的python代碼,對圖像數據集進行批量重命名。同時,舉一反三,可對其他文件類型進行批量重命名

步驟:
1)將需要重命名的圖片文件放入“picture”文件夾;
2)相同路徑下,創建py文件,複製以下代碼;
3)編譯執行,完成批量重命名。

import os

def rename():
    # 原始圖片路徑
    path = r'picture'
    # 獲取該路徑下所有圖片
    filelist = os.listdir(path)
    a = 1
    for files in filelist:
        # 原始路徑
        Olddir = os.path.join(path, files)

        # if os.path.isdir(Olddir):
        #	continue
        # 將圖片名切片,比如 xxx.bmp 切成xxx和.bmp
        # xxx
        filename = os.path.splitext(files)[0]
        # .bmp
        filetype = os.path.splitext(files)[1]
        # 需要存儲的路徑 a 是需要定義修改的文件名
        Newdir = os.path.join(path, str(a) + filetype)
        os.rename(Olddir, Newdir)
        a += 1


rename()

效果

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章