MMtracking训练自己的ReID分类器和检测器

对于像 DeepSORT、Tracktor 这样的 MOT 算法,算法本身由目标检测器ReID分类器两部分构成,因此如果需要做自定义数据集的跟踪,就得先训练一个目标检测器和一个ReID分类器,之后再测试 MOT 整体模型。

A. 训练 ReID分类器

训练ReID分类器有两种办法,第一种是采用MMClassification,另一种是采用MMtracking-1.0.0rc1\configs\reid下的配置文件来训练,二者的区别如下:

  • 采用MMClassification:训练的backbone可选更多且相关的config已经官方已经给出,因此相对于直接采用MMtracking的reid配置文件来训练更加灵活。
  • 采用MMtracking-1.0.0rc1\configs\reid下的配置文件:其配置文件和跟踪的分类器部分的config高度一致,不用做太多修改,训练好直接拿来用就行,比较简

你可能感兴趣的:(目标跟踪干货分享,mmtracking,目标跟踪,训练)