颜色分类leetcode人-属性-识别-MarketDuke在PyTorch中为行人属性识别任务提供了简单基线，实现了对Market-1501-attribute和DukeMTMC-reID-attribute数据集的评估。

数据集

您可以从官方网站下载Market-1501和DukeMTMC-reID数据集。在您的数据集路径下创建一个名为“attribute”的文件夹，并将相应的注释放入该文件夹中。例如：


├── dataset

│ ├── DukeMTMC-reID

│ ├── bounding_box_test

│ ├── bounding_box_train

│ ├── query

│ ├── attribute

│ ├── duke_attribute.mat

模型

项目提供了经过训练的模型，您可以通过相关链接下载，并将checkpoints文件夹移动到项目的根目录。

依赖关系

  • Python 3.5

  • PyTorch >= 0.4.1

  • torchvision >= 0.2.1

  • matplotlib、sklearn、prettytab