Pythonスクリプトを使用してcaffe train_list.txtを生成する方法

以下は caffe train_list.txt を生成するための Python スクリプトを共有する方法です。これは良い参考値なので、皆さんのお役に立てれば幸いです。一緒に見てみましょう

まずコードを与えます:

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 Rename: ファイルとフォルダーは両方とも1つのコマンドです:

os. original_dir,new_dir)

1 ファイル操作:

os.mknod("test.txt") 空のファイルを作成します
fp = open("test.txt",w) ファイルを直接開きます。ファイルが存在しないため、ファイルを作成します

オープンモードについて:

w 書き込みモードで開きます、
a 追加モードで開きます (EOF から開始し、必要に応じて新しいファイルを作成します)
r+ 読み取り/書き込みモードで開きます
w+ 読み書きモードでオープン オープン ( w を参照)
a+ 読み書きモードでオープン ( a を参照)
rb バイナリ読み取りモードでオープン
wb バイナリ書き込みモードでオープン ( w を参照)
ab バイナリ追加モードでオープン ( a を参照)
rb+ バイナリで読み取り書き込みモードでオープン (r+ を参照)
wb+ バイナリ読み取り/書き込みモードでオープン (w+ を参照)
ab+ バイナリ読み取り/書き込みモードでオープン (a+ を参照)

fp.read([size ]）#sizeは読み取りの長さであり、ユニットはbyte

fp.readline（[size]）#read read a lineを読みます。 .readlines([size]) メンバーを取得し、このリストを返します。実際、これはループ内で readline() を呼び出すことによって内部的に実装されます。 size パラメータが指定されている場合、size は読み取られるコンテンツの全長を表します。これは、ファイルの一部のみが読み取られることを意味します。 Tfp.write (STR) #str をファイルに書き込みます。WRITE () は行を追加しません。ffp.writelines (SEQ) #後ろの STR 行はすべて一度に書き込まれます)。この関数も、各行の後に何も追加せずに、忠実に記述するだけです。

fp.close（） Python は、使用されなくなったファイルを自動的に閉じますが、この機能は保証されていません。自分でファイルを閉じる習慣を身に付けることをお勧めします。ファイルが閉じた後に動作している場合、valueRrorが発生します。

fp.next() ファイル操作フラグを次の行に移動します。ファイルが 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:abc.jpg')

'c.jpg'

os.path.commonprefix(list) #リスト内のすべてのパスの共通プレフィックスを返します (複数のパス) ) 最長のパス。
os.path.dirname(path) #ファイルパスを返します
os.path.exists(path) #パスが存在する場合はTrueを返し、パスが破損している場合はFalseを返します
os.path.lexists #パスが存在する場合はTrueを返します、パスが破損している場合は戻ります True
os.path.expanduser(path) #パスに含まれる「~」と「~user」をユーザーディレクトリに変換します
os.path.expandvars(path) #「」を置き換えます環境変数name"と${name}"の値に従ってパスに含まれる$"
os.path.getatime(path) #このパスが最後に入力された時刻を返します。
os.path.getmtime(path) #このパスの最後の変更時刻を返します。
os.path.getctime(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[, ...]] ) #ディレクトリ名とファイル名を1つのパスに結合します
os.path .normcase(path) #パスの大文字と小文字とスラッシュを変換します
os.path.normpath(path) #パス文字列を正規化します
os.path.realpath (path) # path の実パスを返す
os.path.relpath (path[, start]) # start からの相対パスを計算
os.path.samefile(path1, path2) # ディレクトリまたはファイルがSame
os.path.sameopenfile(fp1, fp2) #fp1 と fp2 が同じファイルを指しているかどうかを判断します
os.path.samestat(stat1, stat2) #stat タプル stat1 と stat2 が同じファイルを指しているかどうかを判断します

os.path.split(path) #パスをディレクトリ名とベース名に分割し、タプルを返します ：
os.path.split('E:abc.jpg')

'E:\a\b ','c.jpg'
os.path.splitdrive(path) #通常 Windows で使用され、ドライブ名とパスのタプルを返します
os.path.splitext(path) #パスを分割し、パス名とパスのタプルを返しますファイル拡張子

os.path.split('E:abc.jpg')

'E:\a\b\c','.jpg'

os.path.splitunc(path) #パスをロードポイントとファイルに分割します
os.path.walk(path, visit, arg) #各ディレクトリに入るときにパスをトラバースし、visit 関数を呼び出します

3 つのパラメータ (arg, dirname) が必要です。 dirname は現在のディレクトリのディレクトリ名を表し、names は walk

os.path.supports_unicode_filenames の 3 番目のパラメータです #Unicode パス名をサポートするかどうかを設定します

以上がPythonスクリプトを使用してcaffe train_list.txtを生成する方法の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。