代码笔记:caffe-reid的数据预处理

来源:互联网 发布:时空猎人刷魔晶软件 编辑:程序博客网 时间:2024/06/06 11:48

代码地址如下。该代码是用caffe实现的《A Discriminatively Learned CNN Embedding for Person Re-identification》
https://github.com/D-X-Y/caffe-reid

数据预处理

关于其数据集marker1501的下载及配置,上诉github链接就有详细的说明。而代码中的数据预处理模块是值得重点关注的内容。examples/market1501/mat-codes/generate_train.m

该段matlab代码的实际作用是,将解压后的数据集中‘bounding_box_test’、‘bounding_box_test’和‘query’共3个文件的图像地址、图像名称、图像序号进行读取,并重新存储。
注意:revers_id(trainID(n))-1实现了train_list_file可从0开始遍历。

train_list_file = fopen('../lists/train.lst', 'w');for n = 1:n_train    fprintf(train_list_file, '%s/%s %d\n', train_dir, train_files(n).name, revers_id(trainID(n))-1);    %fprintf(train_list_file, '%s/%s %4d %4d\n', train_dir, train_files(n).name, trainID(n), revers_id(trainID(n)));endfclose(train_list_file);
0 0