源码解析

Person_reID_baseline_pytorch 是行人重识别领域强大的 baseline,基于 pytorch 框架实现,本文尝试对其源码进行解析,方便对行人重识别感兴趣的小伙伴快速了解这个 strong baseline。baseline 的源码链接在 Github 上的 Person_reID_baseline_pytorch

baseline 主要包括如下几个脚本

  • prepare.py 负责数据集的重构
  • model.py 负责网络模型的构建
  • train.py 负责网络模型的训练
  • test.py 负责数据集的特征提取
  • evaluate_gpu.py 负责网络模型的评估
  • demo.py 负责模型效果的简单可视化

之后的文章将详细介绍各脚本的作用,并进行源码解析。

参考文献

  1. Person_reID_baseline_pytorch
  2. 从零开始行人重识别
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐