Skip to content

Latest commit

 

History

History
35 lines (20 loc) · 1.04 KB

README.md

File metadata and controls

35 lines (20 loc) · 1.04 KB

Re-ID MGN模型

Reproduction of paper:Learning Discriminative Features with Multiple Granularities for Person Re-Identification

1. 模型转换

  • 结构文件转换
  • 权重文件转换
  • 预测结果验证
  • 结构文件转换
  • 权重文件转换
  • 预测结果验证

修改源码之后的DarkNet

2. 模型训练

MGN模型的训练,在Pytorch下进行

Remark

(1)在caffe2darknet的转换中,slice层由于DarkNet不支持,需要修改DarkNet源码添加新层实现

(2)需要修改源码让max_pool层支持多尺度kernel size和stride size

(3)修改DarkNet源码,增加了Python接口调用对numpy格式输入的支持