前言:在數據預處理的時候,特別是在將圖像數據轉爲tfrecord格式的時候,有時候類別文件夾名稱不被讀入。需要針對這個問題更改文件夾名稱。這時候就產生了如下工作:
代碼功能:批量重命名文件夾名稱
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import os, sys
path1 = 'E:\\Desktop\\test\\data' #所需修改文件夾所在路徑
dirs = os.listdir(path1)
i = 0
for dir in dirs:
os.rename(str(dir),str(i))
print ("重命名成功!")
i += 1
# 打印出重命名後的目錄
print ("目錄爲: %s" %os.listdir(os.getcwd()))
這裏將文件夾重命名爲0、1、2....
[注]:這裏有點小問題,就是在重新命名的時候.py文件也被重新命名了。不過無傷大雅,後邊有時間再做爲完善。