前言:在用深度學習的方法做目標分類時,會遇到創建數據集的問題。比如說需要創建200類,每類目標存放500張圖片。這樣就需要首先建立200個不同名稱的文件夾。然而,如果一個個創建顯然要不得,因此,就有了以下的代碼:
1.批量創建N個文件夾:
import os, sys
def MkDir():
path = './file/'#創建文件路徑
i = 0
for i in range(200): #創建文件個數
file_name = path + str(i)
os.mkdir(file_name)
i=i+1
MkDir()
2.批量創建N個指定名稱的文件夾:
import os, sys
path1 = './dataset' #指定名稱文件夾所在路徑
path2 = './file/' #新建文件夾所在路徑
def MkDir():
dirs = os.listdir(path1)
for dir in dirs:
file_name = path2 + str(dir)
os.mkdir(file_name)
MkDir()