#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ = "wxf"
# Email: [email protected]
# Date: 2018/8/20

import os

def get_file_path(file_dir):
    file_path_list = []
    for root, dirs, files in os.walk(file_dir):
        # print(u'當前目錄路徑', root)  # 當前目錄路徑
        # print(u'當前路徑下所有子目錄', dirs)  # 當前路徑下所有子目錄
        # print(u'當前路徑下所有非目錄子檔案', files)  # 當前路徑下所有非目錄子檔案
        for file_name in files:
            file_path_list.append(os.path.join(root, file_name))
            # print(os.path.join(root, file_name))
        pass
    pass
    return file_path_list
pass

if __name__ == '__main__':
    file_dir_base = 'G:/wxf/deep_Learn/sphereface/train/data/'
    data_set_name = 'lfw'
    data_set_dir = file_dir_base + data_set_name

    f1 = open(os.path.join(file_dir_base, data_set_name + '.txt'), 'w')
    data_set_path_list = get_file_path(data_set_dir)
    for t in data_set_path_list:
        f1.write(t + '\n')
    pass
    f1.close()
pass