行人重识别项目 | 基于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
相关推荐
Dongsheng_20191 小时前
【汽车篇】AI深度学习在汽车零部件外观检测——机电轴承的应用
人工智能·深度学习·汽车
努力学习的小廉1 小时前
我爱学算法之—— 模拟(下)
c++·算法
en-route2 小时前
从零开始学神经网络——LSTM(长短期记忆网络)
人工智能·深度学习·lstm
海琴烟Sunshine2 小时前
Leetcode 26. 删除有序数组中的重复项
java·算法·leetcode
PAK向日葵3 小时前
【算法导论】NMWQ 0913笔试题
算法·面试
PAK向日葵3 小时前
【算法导论】DJ 0830笔试题题解
算法·面试
PAK向日葵3 小时前
【算法导论】LXHY 0830 笔试题题解
算法·面试
努力毕业的小土博^_^3 小时前
【深度学习|学习笔记】详细讲解一下 深度学习训练过程中 为什么 Momentum 可以加速训练?
人工智能·笔记·深度学习·学习·momentum
清风吹过3 小时前
少样本学习论文分享:多模态和类增量学习
论文阅读·人工智能·深度学习·学习·机器学习
麦麦麦造3 小时前
DeepSeek突然发布 V3.2-exp,长文本能力加强,价格进一步下探
算法