python讀取資料集檔案下所有檔案並打亂劃分生成訓練測試txt檔案(生成train.txt、test.txt,順序隨機,預設比例8:2)
因為每次讀取資料集都要寫載入程式碼,十分浪費時間,而我本人又比較懶。故寫了個python小工具,一鍵執行,讀取資料集檔案下所有檔案並打亂劃分生成訓練測試txt檔案(生成train.txt、test.txt,順序隨機,預設比例8:2),十分方便。
執行指令碼前需要將資料夾整理為這類的形式:
root = “C:\Users\hq\Desktop\HoldingObject\pokemon”(記得改為自己的根路徑)
路徑樹如下(資料夾、圖片數量無限制):
—pokemon
——0(此處資料夾名為標籤)
————1,.jpg、2.jpg、3.jpg、。。。(此處命名無所謂,不影響)
——1 (此處資料夾名為標籤)
————1,.jpg、2.jpg、3.jpg、。。。(此處命名無所謂,不影響)
——2 (此處資料夾名為標籤)
————1,.jpg、2.jpg、3.jpg、。。。(此處命名無所謂,不影響)**
import os
import numpy as np
root = r"C:\Users\hq\Desktop\HoldingObject\pokemon"
#構建所有檔名的列表,dir為label
filename = []
#label = []
dirs = os.listdir(root)
for dir in dirs:
dir_path = root + '\\' + dir
names = os.listdir(dir_path)
for n in names:
filename.append(dir_path + '\\' + n + '\t' + dir)
#打亂檔名列表
np.random.shuffle(filename)
#劃分訓練集、測試集,預設比例4:1
train = filename[:int(len(filename)*0.8)]
test = filename[int(len(filename)*0.8):]
#分別寫入train.txt, test.txt
with open('train.txt', 'w') as f1, open('test.txt', 'w') as f2:
for i in train:
f1.write(i + '\n')
for j in test:
f2.write(j + '\n')
print('成功!')
生成的檔案如下圖所示:
先佔個坑,等我回去補圖。。。
相關文章
- Java 讀取txt檔案生成Word文件Java
- C# 讀取txt檔案生成Word文件C#
- php程式碼生成txt檔案並下載PHP
- python讀取和生成excel檔案PythonExcel
- 遞迴遍歷磁碟下的某一資料夾中所有檔案,並copy檔案生成檔案和帶資料夾的檔案遞迴
- php如何上傳txt檔案,並且讀取txt檔案PHP
- python中按照資料夾中檔案的排列順序讀取檔案內容Python
- Python生成器讀取大檔案Python
- 生成固定大小的隨機檔案隨機
- 讀取本地Excel檔案生成echartsExcelEcharts
- 檔案隨機或順序讀寫原理深入淺出隨機
- python實現word2vec訓練結果bin檔案轉txt檔案Python
- Swift 下載檔案,並讀取Swift
- python生成CSV檔案Python
- 讀取檔案流並寫入檔案流
- python解壓並讀取檔案Python
- 亂序排列檔案中的所有行(轉)
- 生成 api檔案API
- SSIS 生成檔案
- python讀取檔案——python讀取和儲存mat檔案Python
- 檔案中隨機讀取行問題隨機
- python讀取並寫入mat檔案Python
- struts檔案配置順序
- python實現將資料夾內所有txt檔案合併成一個檔案Python
- HDFS讀檔案過程分析:讀取檔案的Block資料BloC
- Linux啟動時讀取配置檔案的順序Linux
- 如何生成csr檔案
- 流式生成Excel檔案Excel
- java 生成 excel檔案JavaExcel
- POI生成EXCEL檔案Excel
- Python解析XML檔案生成HTMLPythonXMLHTML
- shell讀取構建檔案資訊生成json字串JSON字串
- windows下擴充套件yaf,並生成yaf框架檔案Windows套件框架
- 資料庫引數檔案控制檔案日誌檔案資料檔案跟蹤檔案等8大檔案的字典資料庫
- 【自動化測試】Python 讀取 .ini 格式檔案Python
- 批次檢測主機IP,並把結果生成excel檔案表格Excel
- 使用yaml檔案讀取資料YAML
- LIUNUX如何擷取txt檔案中的內容,並建立新檔案UX