下面要為大家分享一篇python腳本產生caffe train_list.txt的方法,具有很好的參考價值,希望對大家有幫助。一起來看看吧
先給程式碼:
import os path = "/home/data//" path_exp = os.path.expanduser(path) classes = [int(p) for p in os.listdir(path_exp)] classes.sort() # nrof_classes一个数据集下有多少个文件夹,就是说有多少个人,多少个类别 nrof_classes = len(classes) count=0 files = open("train_list.txt",'w') filets = open("test_list.txt",'w') count_u=0 for i in range(nrof_classes): class_name = str(classes[i]) count=count+1 count_u=count_u+1 facedir = os.path.join(path_exp, class_name) prefix1 = path+class_name+"/" if os.path.isdir(facedir): images = os.listdir(facedir) #print(images[0]) image_paths = [(prefix1+img+" "+class_name+"\n") for img in images] #print(image_paths[0]) if count < 0.8*nrof_classes: if len(image_paths)>4: test_path=[] for x in range(2): test_path.append(image_paths[0]) del image_paths[0] filets.writelines(test_path) files.writelines(image_paths) #if count==2: # break #imgae_pathses = [] #防止图像大小为0 #for x in image_paths: # if os.path.getsize(x)>0: # imgae_pathses.append(x) #if len(imgae_pathses)==0: # os.rmdir(facedir) files.close() filets.close()
python下os模組的一下有用的用法:
0 重新命名:檔案和資料夾都是一個指令:
os.rename(original_dir,new_dir)
1 檔案動作:
os.mknod("test.txt") 建立空白檔案
fp = open("test.txt",w) 直接開啟一個文件,若檔案不存在則建立檔案
關於open 模式:
##w 以寫入方式打開,a 以追加模式開啟(從EOF 開始, 必要時建立新檔案)
r 以寫入模式開啟
w 以讀寫模式開啟(參見w )
a 以讀寫模式開啟(參見a )
rb 以二進位讀取模式開啟時開啟(參見a )
rb 以二進位讀取模式開啟時開啟 (ffbb )
ab 以二進位追加模式開啟(參見a )
rb 以二元讀寫模式開啟(參見r )
wb 以二元讀取模式開啟(參見w )
ab 以二元讀取模式開啟(參見w )
ab 開啟(參見a )
fp.read([size]) #讀一行,如果定義了size,有可能回傳的只是一行的一部分
fp.readlines([size]) #將文件每一行當作一個list中的一個成員,並且回到這個list。其實它的內部是透過循環呼叫readline()來實現的。如果提供size參數,size是表示讀取內容的總長,也就是說可能只讀到文件的一部分。
fp.write(字符串內容全部寫到文件中(多行一次寫入)。這個函數也只是忠實地寫入,不會在每行後面加上任何東西。
fp.close() #且已關閉檔案。 python會在一個文件不用後自動關閉文件,不過這項功能沒有保證,最好還是養成自己關閉的習慣。如果一個檔案在關閉後也操作會產生ValueError
fp.flush() () 型的」文件標籤「
fp.isatty() 以文件的開頭為原點
fp.next() 把一個file用於for … in file這樣的語句時,就是呼叫next()函數來實現遍歷的。
fp.seek(offset[,whence]) #将文件打操作标记移到offset的位置。这个offset一般是相对于文件的开头来计算的,一般为正数。但如果提供了whence参数就不一定了,whence可以为0表示从头开始计算,1表示以当前位置为原点计算。2表示以文件末尾为原点进行计算。需要注意,如果文件以a或a+的模式打开,每次进行写操作时,文件操作标记会自动返回到文件末尾。
fp.truncate([size]) #把文件裁成规定的大小,默认的是裁到当前文件操作标记的位置。如果size比文件的大小还要大,依据系统的不同可能是不改变文件,也可能是用0把文件补到相应的大小,也可能是以一些随机的内容加上去。
也可以直接打开式的新建文件:
with open('path0.txt','ab') as f: for d in arr: np.savetxt(f,d,fmt='%5f')
path0.txt将直接在python脚本所在目录新建
arr是shape是height x width x channel的numpy数组,注意,这里这么写是因为numpy一次只能够保存一个height x width维度的数据
2 新建、删除文件夹
新建:os.makedirs()
比如windows下新建E:\\dir\\subdir\\
os.makedirs('E:\\dir\\subdir\\') 或者 os.makedirs('E:/dir/subdir/')
ubuntu下新建就是os.makedirs('/home/dir/subdir/')
复制文件:
shutil.copyfile("oldfile","newfile") oldfile和newfile都只能是文件
shutil.copy("oldfile","newfile")oldfile只能是文件夹,newfile可以是文件,也可以是目标目录
复制文件夹:
shutil.copytree("olddir","newdir")olddir和newdir都只能是目录,且newdir必须不存在
重命名文件(目录)
os.rename("oldname","newname") 文件或目录都是使用这条命令
移动文件(目录)
shutil.move("oldpos","newpos")
删除文件
os.remove("file")
删除目录
os.rmdir("dir")只能删除空目录
shutil.rmtree("dir")空目录、有内容的目录都可以删
转换目录
os.chdir("path") 换路径
获取文件大小:os.path.getsize(filename)
3 需要注意的是,在ubuntu下有可能路径包括~,因此需要展开,展开使用:
path_exp = os.path.expanduser(nam)
4 还有一种情况,路径相连,需要将一级一级的目录连接起来,os会自动处理层级目录之间的/,比如,将目录/home 和 数据 1之间连接起来:
os.path.join('/home', str(1)),
后面也可以为文件名,是一样的用法
5 路径存在:用于判断目录是否存在
a = os.path.exists(dir)
存在返回True,否则False
6 获取目录列表,就是给出的目录下的文件或目录列表:
classes = os.listdir(path_exp)
获得的是path_exp下的所有文件的列表,包含目录和文件,
os.path.dirname(path) #返回文件路径,或者是给定路径去掉最后一个目录后的路径,
还有一种使用方式是用在文件中:
os.path.dirname(__file__)用于获取所在文件的路径
7 运行shell命令: os.system():
ubuntu下:os.system(‘pwd')
Python脚本工作的目录路径: os.getcwd()
读取和设置环境变量:os.getenv() 与os.putenv()
给出当前平台使用的行终止符:os.linesep Windows使用'\r\n',Linux使用'\n'而Mac使用'\r'
指示你正在使用的平台:os.name 对于Windows,它是'nt',而对于Linux/Unix用户,它是'posix'
创建多级目录:os.makedirs(r“c:\python\test”)
创建单个目录:os.mkdir(“test”)
获取文件属性:os.stat(file)
修改文件权限与时间戳:os.chmod(file)
终止当前进程:os.exit()
更全面的os.path用法:
os.path.abspath(path) #返回绝对路径
os.path.basename(path) #返回文件名
os.path.basename('E:a\b\c.jpg')
#'c.jpg'
os.path.commonprefix(list) #返回list(多個路徑)中,所有path共有的最長的路徑。
os.path.dirname(path) #返回檔案路徑
os.path.exists(path) #路徑存在則返回True,路徑損壞返回False
os.path.lexists #路徑存在則返回True,路徑損壞也回傳True
os.path.expanduser(path) #把path包含的"~"和"~user"轉換成使用者目錄
os.path.expandvars(path) #根據環境變數的值替換path中包含的”$name”和”${name}”
os.path.getatime(path) #傳回最後一次進入此path的時間。
os.path.getmtime(path) #傳回在此path下最後一次修改的時間。
os.path.getctime(path) #傳回path的大小
os.path.getsize(path) #傳回檔案大小,如果檔案不存在就回傳錯誤
os.path.isabs(path) #判斷是否為絕對路徑
os.path.isfile(path) #判斷路徑是否為檔案
os.path.isdir(path) #判斷路徑是否為目錄
os .path.islink(path) #判斷路徑是否為連結
os.path.ismount(path) #判斷路徑是否為掛載點()
os.path.join(path1[, path2 [, ...]]) #把目錄和檔名合成一個路徑
os.path.normcase(path) #轉換path的大小寫與斜線
os.path.normpath(path ) #規範path字串形式
os.path.realpath(path) #返回path的真實路徑
os.path.relpath(path[, start]) #從start開始計算相對路徑
os .path.samefile(path1, path2) #判斷目錄或檔案是否相同
os.path.sameopenfile(fp1, fp2) #判斷fp1和fp2是否指向相同檔案
os.path.samestat(stat1, stat2 ) #判斷stat tuple stat1和stat2是否指向同一個檔案
os.path.split(path) #把路徑分割成dirname和basename,回傳一個元組:
os.path.split('E:a\b\c.jpg')
#'E:\\a\\b','c.jpg'
os .path.splitdrive(path) #一般用在windows下,傳回磁碟機名稱與路徑組成的元組
os.path.splitext(path) #分割路徑,傳迴路徑名與檔案副檔名的元組
#os.path.split('E:a\b\c.jpg')
'E:\\a\\b\\c ','.jpg'
os.path.splitunc(path) #把路徑分割成載入點與檔案
os.path.walk(path, visit, arg) #遍歷path ,進入每個目錄都呼叫visit函數,visit函數必須有
3個參數(arg, dirname, names),dirname表示目前目錄的目錄名,names代表目前目錄下的所有
檔名,args則為walk的第三個參數
os.path.supports_unicode_filenames #設定是否支援unicode路徑名
#
以上是python腳本產生caffe train_list.txt的方法的詳細內容。更多資訊請關注PHP中文網其他相關文章!