按照lfw pairs.txt的格式生成自己的pairs.txt

1、把自己的数据集按照官方命名格式修改

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Fri Oct 18 11:29:55 2019
change the name of the image
@author: cong
"""import os
#import re
imgs_path = '/home/cong/facenet/dataset/foda_imgs'if os.path.isdir(imgs_path):for s in os.listdir(imgs_path):newdir=os.path.join(imgs_path,s)#print(newdir)filenames = os.listdir(newdir)filenames.sort()for name in filenames:newname = newdir.split('/')[-1]os.rename(newdir + '/' + name, os.path.join(newdir, newname + '_' + '%04d' % int(filenames.index(name)+1)) + '.png')#print(newname)

2、生成自己的pairs.txt。

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Fri Oct 18 13:38:56 2019@author: cong
"""import glob
import os.path
import numpy as np
import os
from itertools import combinations
# 图片数据文件夹
INPUT_DATA = '/home/cong/facenet/dataset/foda_imgs'def create_image_lists():matched_result = set()k = 0# 获取当前目录下所有的子目录,这里x 是一个三元组(root,dirs,files)，第一个元素表示INPUT_DATA当前目录，# 第二个元素表示当前目录下的所有子目录,第三个元素表示当前目录下的所有的文件sub_dirs = [x[0] for x in os.walk(INPUT_DATA)]for sub_dir in sub_dirs[1:]:# 获取当前目录下所有的有效图片文件extensions = 'png'# 把图片存放在file_list列表里file_list = []# os.path.basename(sub_dir)返回sub_sir最后的文件名file0_list = []dir_name = os.path.basename(sub_dir)file_glob = os.path.join(INPUT_DATA, dir_name, '*.' + extensions)# glob.glob(file_glob)获取指定目录下的所有图片，存放在file_list中file_list.extend(glob.glob(file_glob))file0_list.extend(glob.glob(file_glob))file0_list.sort()# 通过目录名获取类别的名称label_name = dir_name#print(label_name)length = len(file_list)d_com = []d = np.arange(length)for i,j in combinations(d, 2):d_com.append([i,j])base_name1 = file0_list.index(file_list[i])+1  # 获取文件的名称base_name2 = file0_list.index(file_list[j])+1s = label_name+'\t'+str(base_name1)+'\t'+str(base_name2)if s not in matched_result:# 将当前类别的数据放入结果字典matched_result.add(label_name +'\t'+ str(base_name1)+ '\t'+ str(base_name2))k = k + 1#print(len(matched_result))# 返回整理好的所有数据return matched_result, kdef create_pairs():unmatched_result = set()       # 不同类的匹配对k = 0sub_dirs = [x[0] for x in os.walk(INPUT_DATA)]# sub_dirs[0]表示当前文件夹本身的地址，不予考虑，只考虑他的子目录for sub_dir in sub_dirs[1:]:# 获取当前目录下所有的有效图片文件extensions = ['png']file_list = []# 把图片存放在file_list列表里dir_name = os.path.basename(sub_dir)for extension in extensions:file_glob = os.path.join(INPUT_DATA, dir_name, '*.'+extension)# glob.glob(file_glob)获取指定目录下的所有图片，存放在file_list中file_list.extend(glob.glob(file_glob))length_of_dir = len(sub_dirs)a = np.arange(1,length_of_dir)a_com = []for i,j in combinations(a, 2):#print(i,j)a_com.append([i,j])class1 = sub_dirs[i]class2 = sub_dirs[j]# print(class1,'\n',class2)class1_name = os.path.basename(class1)class2_name = os.path.basename(class2)# 获取当前目录下所有的有效图片文件extensions = 'png'file_list1 = []file_list11 = []file_list2 = []file_list22 = []# 把图片存放在file_list列表里file_glob1 = os.path.join(INPUT_DATA, class1_name, '*.' + extension)file_list1.extend(glob.glob(file_glob1))file_list11.extend(glob.glob(file_glob1))file_list11.sort()file_glob2 = os.path.join(INPUT_DATA, class2_name, '*.' + extension)file_list2.extend(glob.glob(file_glob2))file_list22.extend(glob.glob(file_glob2))file_list22.sort()len_of_file = len(file_list1)if file_list1 and file_list2:b = np.arange(len_of_file)b_com = []for m,n in combinations(b, 2):#print(m,n)b_com.append([m,n])base_name1 = file_list11.index(file_list1[m % len(file_list1)])+1  # 获取文件的名称base_name2 = file_list22.index(file_list2[n % len(file_list2)])+1# unmatched_result.add([class1_name, base_name1, class2_name, base_name2])s = class1_name+'\t'+str(base_name1)+'\t'+class2_name+'\t'+str(base_name2)if s not in unmatched_result:unmatched_result.add(class1_name+'\t'+str(base_name1)+'\t'+ class2_name+'\t'+str(base_name2))k = k + 1return unmatched_result, kresult, k1 = create_image_lists()
print(len(result))
# print(result)result_un, k2 = create_pairs()
print(len(result_un))
# print(result_un)file = open('/home/cong/facenet/data/pairs_foda.txt', 'w')result1 = list(result)
result2 = list(result_un)file.write('10 300\n')j = 0
for i in range(10):j = 0print("=============================================第" + str(i) + '次, 相同的')for pair in result1[i*200:i*200+200]:j = j + 1print(str(j) + ': ' + pair)file.write(pair + '\n')print("=============================================第" + str(i) + '次, 不同的')for pair in result2[i*200:i*200+200]:j = j + 1print(str(j) + ': ' + pair)file.write(pair + '\n')

按照lfw pairs.txt的格式生成自己的pairs.txt相关推荐

html文件转为txt,html转txt文件怎么实现？html转txt的方法详解！
HTML转TXT很简单,需要一台计算机即可操作,只需要把网页HTML文档的扩展名.html改成TXT文本格式的扩展名,即可用记事本打开. 如果没有显示扩展名的需要计算机设置显示扩展名. 下面我们介绍如 ...
3D物体分类---ModelNet .OFF格式生成多视角图像（Blender方法之第二步）
3D物体分类---ModelNet .OFF格式生成多视角图像(Blender方法之第二步) 强烈建议使用Blender方法!!! 说明 2 python调用Blender生成多视角图像 2.1 Bl ...
python 将YOLO（txt）格式的标注数据批量转换为PascalVOC（XML）格式的标注数据
python 将YOLO(txt)格式的标注数据批量转换为PascalVOC(XML)格式的标注数据准备工作修改代码路径运行代码程序bug(没时间看所以还没解决): 准备工作需在目标文件夹中 ...
yolo图像检测数据集格式转换：xml 与 txt格式相互转换
格式介绍一图流介绍的比较详细,一般图像检测数据集格式为txt或者xml格式,在使用labelimg进行标注的时候,可以设置获得不同格式的数据集,以满足不同算法训练格式要求: 一般建议使用pascal ...
Python重装时记录所有库生成requirement 使用requirements.txt
Python 生成requirement 使用requirements.txt python项目中必须包含一个 requirements.txt 文件,用于记录所有依赖包及其精确的版本号.以便新环境部 ...
(原创)按照一定的格式生成一定数量的随机数的例子
在很多时候需要按照一定的格式生成一定数量的随机数,下面是实现方法下面是自己写的类 public class MyRandom { private string _strin; privat ...
计算机二级目录设置,word2设置标题格式,生成目录,奇偶页设置等等,适用考计算机二级办公软件,也适用于毕业论文格式设置...
word2设置标题格式,生成目录,奇偶页设置等等,适用考计算机二级办公软件,也适用于毕业论文格式设置这是第2部分word资料 , 跟之前的第一部份word资料合为一份第二题 2.在正文前按序插入节 ...
java 9999 符号_按格式生成序号，如0001,0002...9999
/*** * 按格式生成序号,如0001,0002...9999 * @param idx 序号号码 * @param scale 位数 * @return 按位数格式化的序号,如0019 */ pu ...
python图片保存为txt文件_python实现对文件中图片生成带标签的txt文件方法
在深度学习中经常需要生成带标签的图片名称列表,xxxlist.txt文件,下面写一个简单的python脚本生成该文件列表. import os def generate(dir,label): fil ...

按照lfw pairs.txt的格式生成自己的pairs.txt

按照lfw pairs.txt的格式生成自己的pairs.txt相关推荐

最新文章

热门文章