python:比较人脸识别中gallery数据集和prob数据的IP

1）无论是自己设置的gallery和prob集合，或者网上下载的都有肯能出现偏差，比如本人遇到的:下载的的gallery有355个IP，但是在prob中有362的IP，这样在跑人脸rank1或者验证的时候是非常致命的问题。2）所以自己写了一个简单的python脚本比较两个txt文件中的IP
3）两个txt文件截图如下：
gallery.txt：
s2_VIS_10153_003.jpg
s2_VIS_10117_005.jpg
s3_VIS_20464_004.jpg
s2_VIS_10113_006.jpg
s2_VIS_10136_008.jpg

prob.txt:
s2_NIR_10153_001.jpg
s2_NIR_10153_002.jpg
s2_NIR_10153_003.jpg
s2_NIR_10153_004.jpg
s2_NIR_10153_005.jpg

4)代码如下：思想就是遍历两个txt文件，放入两个list中，根据图片名字进行比较。

import os
import sys

if name == “main”:

f1 = open('gallery1.txt','r') f2 = open('prob1.txt','r') c1 = [] c2 = [] list1 = f1.read().splitlines() for v in list1:#print(v)c1.append(v[0:2]+v[7:12]+' '+v[21:24]) #这里存入你要比较的字符符串 #print(c1) #print(len(c1)) list2 = f2.read().splitlines() for i in list2:c2.append(i[0:2]+i[7:12]+' '+i[21:24]) #这里存入你要比较的字符串result1 = [x for x in c2 if x not in c1] #在gallery中没有，但在prob中存在 print("result1 = ",result1) result2 = [y for y in c1 if y not in c2] #在prob中有，但在gallery中存在 print("result2 = ",result2)

总结

以上是生活随笔为你收集整理的python:比较人脸识别中gallery数据集和prob数据的IP的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。