deep-person-reid源码解析
1.简介
deep-person-reid是一个针对Person re-identification问题的训练框架。它基于深度学习框架PyTorch,提供了规范的代码组织结构,适用于reid问题的实验拓展。
项目gituhb仓库地址:https://github.com/KaiyangZhou/deep-person-reid
项目文档:https://kaiyangzhou.github.io/deep-person-reid/
2.项目源码主要目录
项目的根目录主要由以下文件或文件夹构成
├── configs # 存放训练配置文件(*使用.yaml)
├── docs # 项目文档
├── projects # 作者的论文的源码
│ ├── OSNet_AIN
│ └── README.md
├── scripts # 脚本目录
│ ├── default_config.py
│ └── main.py
├── setup.py # 安装脚本
└── torchreid # torchreid核心代码
├── data
├── engine
├── __init__.py
├── losses
├── metrics
├── models
├── optim
└── utils
3.根目录主要内容解析
3.1 ./configs
目录
该目录包含每个实验的.yaml
配置文件,主要包含model, data, loss, train, test
等方面的设置。该文件被解析成yacs库中的CfgNode
类(类似于easydict对象)。
以im_r50_softmax_256x128_amsgrad.yaml为例:
<