背景:深度學習圖像分類的時候,有許多的标簽類。1000個标簽就需要創建1000個文件夾,文件夾裡面放入對應的圖像。不可能一個一個右擊創建文件夾,所以有了今天的腳本。根據txt文本每一行的類别,創建文件名為類别的文件夾。
# 目标是讀取txt的每一行數據創建對應名字的文件夾
# 1通過函數調用解決創建固定的文件夾
# 2創建一個txt文件輸出裡面的内容輸出
# 3将txt輸出内容和創建文件夾鍊接完成任務
# 4後續補充檢測文本文件是否存在,不在報錯
# 5後續補充如果已經有了文本中的文件夾跳過這個文件名創建下一個
# 結束
# 反思:目前創建的文件夾目錄固定、文本文檔位置固定
# 完善功能:
# 1.先把文件夾目錄改為交互式用戶設置
# 2.檢測txt文件是否存在
# !/usr/bin/python
# -*-coding:utf-8-*-
import os
def mkdir(path):
folder=os.path.exists(path)
if not 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)
Top進入s10 四強,希望今年能捧冠
,更多精彩资讯请关注tft每日頭條,我们将持续为您更新最新资讯!