目標效果:
--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內容中的名字沒有改變,因此導入至模型時需要注意,避免輸入數據讀取報錯