基于表征学习的ReID代码实践

基于表征学习的ReID方法

表征学习(representation learning)是一类常用的行人重识别方法。由于CNN可自动从原始图像数据中根据任务需求自动提取表征特征(Representation),所以有些researcher会把行人重识别问题看作分类(Classification/Identification)问题或者验证(Verification)问题:

  • (1)分类问题是指利用行人的ID或者属性等作为训练标签来训练模型;
  • (2)验证问题是指输入一对(两张)行人图片,让网络来学习两张图片是否属于同一个人。
    如今依然有大量工作是基于表征学习,表征学习也成为了ReID领域的一个非常重要的baseline,并且表征学习的方法比较鲁棒,训练比较稳定,结果也比较容易复现。
    在这里插入图片描述
    表征学习的局限性:容易在数据集的domain上过拟合,并且当训练ID增加到一定程度的时候会显得比较乏力。

代码实现

主要代码

  • data_manager.py: 自动处理数据集,返回数据集的一些常用属性
  • data_loader: 吐数据的dataloader
  • transform.py:数据预处理代码,可以自定义数据增广
  • model.py:提取特征的模型(ResNet50)
  • train.py:主文件,包含训练、测试等代码
  • utils.py:一些可能用到的功能代码
发布了134 篇原创文章 · 获赞 38 · 访问量 9万+

猜你喜欢

转载自blog.csdn.net/rytyy/article/details/105588045