圖像處理項目——生成csv文件提高讀取效率

利用pyhton腳本生成csv文件

*開發環境為windows PyCharm
*使用的是pyhton腳本
*生成人臉和人臉對應的標簽的csv文件

一：主要步驟

1、載入對應路徑
2、提取每一張圖片對應的位置與同一個人臉對應的標簽
3、寫入存儲文件夾data.txt

二：代碼

"""------------------------------------------------------------------
利用pyhton腳本生成csv文件
    人臉模型訓時需要讀取人臉和人臉對應的標簽，利用csv文件讀取,以取代傳統
的逐一的方式,直接在數據庫讀取效率很低。在訓練之前生成csv文件,主要步驟:
1、載入對應路徑
2、提取每一張圖片對應的位置與同一個人臉對應的標簽
3、寫入存儲文件夾data.txt
---------------------------------------------------------------------"""
import tensorflow as tf
import os.path
"""定義csv文件生成函數"""
def MakeLabel():
    face_path = "faces"               # 人臉數據的路徑
    separator = ";"                   # 分隔符
    file = open("data.txt", 'w')      # 打開存儲的文件
    """filenames對應faces; dirnames對應s1; dirname對應1.pgm；其中faces/s1/1.pgm"""
    for dirname, dirnames, filenames in os.walk(face_path):
        for targetname in dirnames:
            target_path = os.path.join(dirname, targetname)
            for filename in os.listdir(target_path):
                targetname2 = targetname[::-1]
                targetname3 = targetname2[:-1]
                targetname4 = targetname3[::-1]

                label = int(targetname4)            # 文件名字符串型轉數字int型
                path = "%s/%s" % (target_path, filename)
                print("%s%s%d" % (path, separator, label))
                file.write(path)                    # 寫入路徑
                file.write(separator)               # 寫入;號
                file.write(str(label))              # 寫入標簽
                file.write("\n")
            label = label + 1
    file.close()                      #關閉存儲的文件

"""啟動函數"""
def main(argv=None):
    MakeLabel()
if __name__ =='__main__':
    tf.app.run()

三 ： 生成結果