python手寫lfw數(shù)據(jù)集轉(zhuǎn)pair.txt形式

最近在搞facenet的人臉識別，用到了lfw數(shù)據(jù)集，發(fā)現(xiàn)沒有個合適的將自己數(shù)據(jù)集轉(zhuǎn)為lfw_pair.txt的代碼，所以自己擼了一個。

數(shù)據(jù)集用的就是lfw，文件夾是名字，名字文件夾內(nèi)放的就是人臉圖像。沒有其他文件所有沒有做是否是圖像的校驗。

主要思路就是根據(jù)lfw的要求，匹配數(shù)據(jù)通過同一個人有多張人臉構(gòu)造，不匹配數(shù)據(jù)通過不同文件夾的內(nèi)容構(gòu)造。

利用python的set數(shù)據(jù)結(jié)構(gòu)去重特性，一直往里面加就行了，加到數(shù)量夠了就break。

然后將兩個set集合都轉(zhuǎn)為list，根據(jù)順序構(gòu)造lfw_pair.txt，構(gòu)造時參考了原來的排序方式，做了sorted。

歡迎批評指教喲

# -*- coding: utf-8 -*- # @Time : 2022/3/21 14:39 # @Author : ChangXinliang # @Software: PyCharmimport os.path import random import osdef create_matched_result(lfw_dir, all_number):matched_result = set()# 獲取所有數(shù)據(jù)名字names = os.listdir(lfw_dir)while len(matched_result) < all_number:for name in names:length = len(os.listdir(os.path.join(lfw_dir, name)))if length <= 1:continueelse:# 構(gòu)造兩個隨機數(shù)number_1, number_2 = sorted(random.sample(range(1, length + 1), 2))s = name + '\t' + str(number_1) + '\t' + str(number_2)# 沒有必要做判斷，set會自行處理matched_result.add(s)if len(matched_result) == all_number:breakreturn list(matched_result)def create_unmatched_result(lfw_dir, all_number):unmatched_result = set()names = os.listdir(lfw_dir)while len(unmatched_result) < all_number:# 隨機取兩個名字name1, name2 = sorted(random.sample(names, 2))numbers = []for name in [name1, name2]:images = os.listdir(os.path.join(lfw_dir, name))# 隨機取人臉number = random.sample(range(1, len(images) + 1), 1)[0]numbers.append(number)# 沒有必要做判斷，set會自行處理s = name1 + '\t' + str(numbers[0]) + '\t' + name2 + '\t' + str(numbers[1])unmatched_result.add(s)return list(unmatched_result)if __name__ == '__main__':# 圖片數(shù)據(jù)文件夾lfw_dir = 'lfw'# 重復(fù)次數(shù)repeat_count = 10# 每次匹配數(shù)量pair_number = 300# 總數(shù)量all_number = repeat_count * pair_number# 構(gòu)造同一張人臉的集合matched_result = create_matched_result(lfw_dir, all_number)# 構(gòu)造不同人臉的集合unmatched_result = create_unmatched_result(lfw_dir, all_number)out_path = r'lfw_pair.txt'with open(out_path, 'w') as f:f.write('%d %d\n' % (repeat_count, pair_number))for i in range(repeat_count):for pair in sorted(matched_result[i * pair_number: (i + 1) * pair_number]):f.write(pair + '\n')for pair in sorted(unmatched_result[i * pair_number: (i + 1) * pair_number]):f.write(pair + '\n')

總結(jié)

以上是生活随笔為你收集整理的python手写lfw数据集转pair.txt形式的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。