1. 将自己的数据集转换为coco类数据集

1. 写一个config文件，可以模仿byteTrack自带的，在exps/example/mot中，主要是修改类别数、以及数据读取路径，其它参数比如normalize值等可以自己修改

1. 训练就好了，标注格式正确，止步于此

1. 如果有些问题，比如 device-side assert triggered，就很可能和label有关，下面是我的数据集coco格式中的categories不正确带来的一系列问题

1. 问题的产生：annotations中的json文件中categories存放的是类别和id的对应情况，而不是每个GT的类别，导致sorted(self.coco.getCatIds())=[1,1,1,……,last_cls_id]

1. 问题解决：显然，annotations中的json文件中categories存放是类别和id的对应情况

1. 问题探索：第5部分是初步解决办法，之后，查看pycocotools源码，见cocoapi/coco.py at master · cocodataset/cocoapi (github.com)，知道self.coco.getCatIds()的功能，就知道json文件中的categories格式了

1. 曲折的探索（让代码适应我的标注格式）

• 在yolox/data/datasets/mot.py中，load_anno_from_ids函数：

        for ix, obj in enumerate(objs): #在77行左右cls = self.class_ids.index(obj["category_id"])res[ix, 0:4] = obj["clean_bbox"]res[ix, 4] = clsres[ix, 5] = obj["track_id"]

以及在__init__()函数中，可以看到：

self.class_ids = sorted(self.coco.getCatIds())

看起来很正常对吧，但是在自己的数据集上，这样获取的cls（目标的类别）是错误的！！！

关于pycocotools的一些应用可以见：

pycocotools一些api的用法记录 - One Blog | 文鹃阁 (szukevin.site)

其中提到了coco.getCatIds()会返回所有类别，并且每个类别只出现一次。在本人使用的数据集上，self.coco.getCatIds会出现所有类别，但是每个类别出现的次数不止一次，因此只是简单的把数据中出现的GT的类别简单读取，并没有去重。

再看load_anno_from_ids中的self.class_ids.index(obj["category_id"])得到当前obj的cat_id的index。所以一开始我直接将self.class_ids.index(obj["category_id"])改成cls = obj["category_id"] - 1，可以正常训练了，不会报错： device-side assert triggered ，你以为正常了吗？No，还不是正常的！你会发现训练了一个epoch之后AP值很低，比如:）

• 在yolox/data/evaluators/coco_evaluator.py以及yolox/data/evaluators/mot_evaluator.py的convert_to_coco_format函数中：

label = self.dataloader.dataset.class_ids[int(cls[ind])]

该函数用于获取检测结果之后，将检测结果转换为coco数据格式，我们前面只是改了读取GT的label。由于self.dataloader.dataset.class_ids所有GT的cls（从1开始）的排序，那么一般是[1,1,1,1,1,……],如果类别为1的GT很多，而int(cls[ind])为range(0,num_class)范围，则得到检测的label就都是1了，指标能不低吗┭┮﹏┭┮，都是1！！！

• 最后的最后

综合考虑，检测结果和GT的label。放弃之前的修改，将yolox/data/datasets/mot.py中，load_anno_from_ids函数改成

        # self.class_ids = sorted(self.coco.getCatIds())self.class_ids = [1,2,3,4,5,6,7,8,9,10,11]#这是我数据集的类别

指标正常，返回检测类别也正常，呜呜

也可以改成如下，多一个去重操作，没试过，理论是可以的。

self.class_ids = sorted(set(self.coco.getCatIds()))

所以·，数据集标注和我一样问题的，可以改改代码使用，但是建议规范数据集吧，太难了呜呜