因为最近开始接触人脸识别,正在拜读林倞老师的《cross domain visual matching via generalized similarity measure and feature learning》,并开始跑这篇论文的caffe代码。由于代码里没有给出cuhk03数据集,以及train_cuhk_domain_1.txt和train_cuhk_domain_2.txt文件,所以我先下载了cuhk03数据集,而且是PNG格式。
然后先在txt文件了写了几行作为测试,发现caffe报错并提示找不到1_001_1_1.jpg。所以我在想是不是林老师的课题组编的底层输入数据的代码只识别jpg格式的图片,另外图片的命名也有一定的讲究。
一、读取文件夹的图片并把图片名写入txt文件
# encoding: UTF-8
import os
import re
def createFileList(images_path, txt_save_path):
# 打开图片列表清单txt文件
fw = open(txt_save_path, "w")
# 查看图片目录下的文件,相当于shell指令ls
images_name = os.listdir(images_path)
# 遍历所有文件名
for eachname in images_name:
# 按照规则将内容写入txt文件中
fw.write(eachname+'\n')
# 打印成功信息
print "生成txt文件成功"
# 关闭fw
fw.close()
# 下面是相关变量定义的路径
if __name__ &