基于paddlle架构的双摄像头下的行人重识别
主要使用的是paddle-class和paddle-detection这两个模块;
源代码仓库如下:
paddle-class
paddle-detection
行人重识别的原理图如下所示:
最后实现的效果,如下所示:
vis_c003
vis_c004
所以我们需要先训练,得到我们想要的REID模型。
reid模型在paddle-clas里面训练,需要的环境如下所示:
prettytable
ujson
opencv-python==4.6.0.66
pillow
tqdm
PyYAML>=5.1
visualdl>=2.2.0
scipy>=1.0.0
scikit-learn>=0.21.0
gast==0.3.3
faiss-cpu
easydict
也可以直接先下载paddle-class的源码仓库,然后使用
pip install -r requirements.txt上面这个命令,进行环境安装。
训练REID模型,如果本地配置不行,大家可以参考我在BML上面公开的项目
paddle-REID
我使用的数据集是DukeMTMC行人重识别数据集;
使用paddleclas完成行人重识别的模型训练 - 飞桨AI Studio星河社区
