行人重识别项目 | 基于Pytorch实现ReID行人重识别算法

  • 项目应用场景
    • 面向行人重识别场景,项目具有轻量化 (训练的时候也只需要 2GB 的显存占用)、性能好 (只使用 softmax 损失就能够达到 Rank@1=88.24%, mAP=70.68%),另外提供友好的上手项目流程教程
  • 项目效果
  • 项目流程 ==> 具体参见项目内README.md

(1) 安装依赖

bash 复制代码
git clone https://github.com/pytorch/vision
cd vision
python setup.py install

git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cuda_ext --cpp_ext

(2) 数据集准备

bash 复制代码
pip install gdown 
pip install --upgrade gdown #!!important!!
gdown 0B8-rUzbwVRk0c054eEozWG9COHM

python prepare.py

gdown 1jjE85dRCMOgRtvJ5RQV9-Afs-2_5dY3O

(3) 执行训练

bash 复制代码
python train.py --gpu_ids 0 --name ft_ResNet50 --train_all --batchsize 32  --data_dir your_data_path

(4) 执行测试

bash 复制代码
python test.py --gpu_ids 0 --name ft_ResNet50 --test_dir your_data_path  --batchsize 32 --which_epoch 59
相关推荐
学编程的小程3 小时前
LeetCode216
算法·深度优先
leeyayai_xixihah3 小时前
2.21力扣-回溯组合
算法·leetcode·职场和发展
01_3 小时前
力扣hot100——相交,回文链表
算法·leetcode·链表·双指针
萌の鱼3 小时前
leetcode 2826. 将三个组排序
数据结构·c++·算法·leetcode
Buling_03 小时前
算法-哈希表篇08-四数之和
数据结构·算法·散列表
AllowM3 小时前
【LeetCode Hot100】除自身以外数组的乘积|左右乘积列表,Java实现!图解+代码,小白也能秒懂!
java·算法·leetcode
RAN_PAND4 小时前
STL介绍1:vector、pair、string、queue、map
开发语言·c++·算法
boooo_hhh4 小时前
深度学习笔记16-VGG-16算法-Pytorch实现人脸识别
pytorch·深度学习·机器学习
fai厅的秃头姐!6 小时前
C语言03
c语言·数据结构·算法
lisanndesu6 小时前
动态规划
算法·动态规划