此存储库包含CVPR'20作品的源代码,带有状态信息的弱监督区分特征学习,用于人员识别(。我们的实现基于Pytorch。以下是使用该代码来训练和评估我们的模型的说明。 先决条件 Python 3.6+ Python软件包:numpy,scipy,pyyaml / yaml,h5py 数据和预先训练的权重 请从密码为“ tih8”的Market / Duke / Multi-PIE / CFP数据集以及预训练的ResNet50权重。 或者,从找到与RE-ID相关的东西,从与面部识别相关的东西。 将所有它们(数据集和预训练的权重)放入/ data / (请在根目录中创建一个文件夹/ data)。 运行代码 请进入主文件夹,然后运行 python src/main.py --gpu 0,1,2,3 --save_path runs/market 在市场数据集上,其中“ 0,1,2,3”指定