python子文件夾內文件排序(加前綴的形式)

目標效果:

--root_dir

   ---dir-1

         001_xxx.jpg, 002_xxx.jpg,...

   ---dir-2

         001_xxx.jpg, 002_xxx.jpg,...

import os

data_path = r"/data/dataset/xxx/naru_region_data/"
dir_list = sorted(os.listdir(data_path))
for idx, dirname in  enumerate(dir_list):
    filename_list = os.listdir(os.path.join(data_path, dirname))
    prefix = "{:0>3d}-".format(idx + 1)
    for filename in filename_list:
        os.renames(os.path.join(data_path, dirname, filename), os.path.join(data_path, dirname, prefix + filename))

問題分析:遇到不是文件夾的文件會報錯,可以在dir_list中優化一下,這裏省事直接把報錯文件刪掉(用來檢查數據集雜樣本)

最終效果還行,但是要注意的是,xml和jpg的名字雖然改了,但是xml內容中的名字沒有改變,因此導入至模型時需要注意,避免輸入數據讀取報錯

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