AAAI顶会行人重识别算法源码解读——Relation Network for Person Re-identification

数据及代码链接见文末

1.项目配置与数据集介绍

        在这里我们使用的是清华大学的行人重识别数据集market1501。数据集的目录如下,images文件夹下是所有的行人图片。poses文件夹下是姿态估计的标签文件(有行人重识别的研究首先进行姿态估计,然后再获得行人的特征表示)。meta.json文件夹下是存放的每个人在6组摄像头的图片。splits文件夹下是数据集的划分方式。json在构建数据集dataloader时需要使用

 image文件夹下,图片的命名规则如下,如00000000_0001_00000011。00000000代表的是行人id.0001代表的是6组摄像头的摄像头id。00000011代表的是这组摄像头下图片的id。因此,我们根据图片名字即可获取标签,而不需要额外的标签文件

2.dataloader加载数据

        行人重识别由于需要计算三元组损失,因此不能随机的加载数据。首先,在取数据上,随机选择一个人,从这个人的所有图片中选择四张图片,构成了一个batch,然后对其中的图片进行数据增强。即一个batch代表的是一个人的四张图片,batch_size为16即代表16个人,共64张图片。

数据选取部分代码如下:

class TripletBatchSampler(S

猜你喜欢

转载自blog.csdn.net/qq_52053775/article/details/129344235