Python 提取資料夾下所有檔案

2021-08-29 13:58:24 字數 1106 閱讀 4270

#tensorflow版faster rcnn 訓練自己的資料集

#需要提取名稱構成訓練集(train.txt)、測試集(val.txt)、驗證集

#time: 2018-12-29 10:50

#@author: moli99

# coding = utf-8

import os

def file_name(file_dir):

#儲存目錄下的所有jpg檔案到l列表

l=

for root, dirs, files in os.walk(file_dir):

files.sort() #對檔案進行排序

# l中儲存的是所有的去除字尾名的jpg

l = file_name(root + '/jpegimages')

# 建立trainandval.txt檔案,寫入的名稱構成訓練測試集

f = open(root + '/imagesets/trainandval.txt','w')

for i in l:

f.write(i + '\n') #寫入

f.close()

print(l)

#手動分割後,形成兩個檔案train.txt和val.txt

#對檔案中順序隨機打亂

import random

files = root + '/imagesets/val.txt'

f = open(files,'r+')

readf = f.readlines()

random.shuffle(readf) #隨機排序

f.seek(0) #檔案指標回0

f.truncate() #清空檔案

for i in readf:

f.write(i) #寫入

f.close()

print readf

python 統計資料夾下的所有資料夾或檔案數目

統計 home jiangt下資料夾的個數 import os path home jiangt count 0 for file in os.listdir path file 表示的是檔名 count count 1 print count獲取資料夾下的檔案的個數 import os path ...

Python遍歷資料夾下所有檔案

7只遍歷當前資料夾 不遞迴遍歷 import glob dir test samples glob.glob dir print samples print len samples test 1.txt test new 2遍歷當前資料夾下所有的.txt import glob dir test s...

python檢視資料夾下所有檔案

實現檢視所有檔案,重點在於資料夾下又有資料夾時怎樣處理,這裡通過os模組來解決.方法一 通過遞迴實現遍歷所有資料夾 import os def func path for i in os.listdir path path2 os.path.join path,i 拼接絕對路徑 if os.path...