【Torchreid】行人重识别deep-person-reid源码测试（一）

一、简介

deep-person-reid是一个针对Person re-identification问题的训练框架。它基于深度学习框架PyTorch，提供了规范的代码组织结构，适用于reid问题的实验拓展。本文就自己第一次训练测试过程进行记录。
项目gituhb仓库地址：https://github.com/KaiyangZhou/deep-person-reid
项目文档：https://kaiyangzhou.github.io/deep-person-reid/

二、环境和数据集准备

2.1 从github上下载源码到本地服务器

cd deep-person-reid-master

2.2 conda创建虚拟环境

conda create --name torchreid python=3.7

2.3 激活虚拟环境

conda activate torchreid

2.3 安装需求包

之后如果运行时还缺其他包再pip安装

pip install -r requirements.txt

2.4 安装pythorch

根据自己的配置安装对应torch版本
在终端输入nvcc -V查看自己的cuda版本。如我是10.2

上官网找到pythorch安装命令
如我安装的pytorch1.8.0版本，使用命令即可安装。

conda install pytorch1.8.0 torchvision0.9.0 torchaudio==0.8.0
cudatoolkit=10.2 -c pytorch