背景:深度學習圖像分類的時候,有許多的標籤類。1000個標籤就需要創建1000個文件夾,文件夾裡面放入對應的圖像。不可能一個一個右擊創建文件夾,所以有了今天的腳本。根據txt文本每一行的類別,創建文件名為類別的文件夾。
# 目標是讀取txt的每一行數據創建對應名字的文件夾# 1通過函數調用解決創建固定的文件夾# 2創建一個txt文件輸出裡面的內容輸出# 3將txt輸出內容和創建文件夾連結完成任務# 4後續補充檢測文本文件是否存在,不在報錯# 5後續補充如果已經有了文本中的文件夾跳過這個文件名創建下一個# 結束# 反思:目前創建的文件夾目錄固定、文本文檔位置固定# 完善功能:# 1.先把文件夾目錄改為交互式用戶設置# 2.檢測txt文件是否存在# !/usr/bin/python# -*-coding:utf-8-*-import osdefmkdir(path):folder=os.path.exists(path) ifnot folder: os.makedirs(path) print(path+"---OK---") else: print(path+"---There is this folder!---") if __name__=='__main__': file = "D:\db\images" with open('labels.txt','r', encoding='utf-8') as f: lines=f.readlines() for line in lines: #folder=file+line folder=line #strip()方法移除字符串頭尾指定的字符 folder=folder.strip() mkdir(folder)