gen_labels

[ymRain]

本人小白一枚，请问数据集为MOT类型，有images和gt.txt文件，需要进行多类别追踪训练，生成labels时，需要对gen_labels_mot16_car.py进行哪些方面的修改呢？提前谢谢大佬解答！

[CaptainEven]

@ymRain 貌似MOT16数据集中的Vehicle数量很少，几乎不能当作MCMOT的训练集。

[ymRain]

我有别的数据集，需要对好多类进行检测，利用gen_labels_mot16_car.py生成labels时，需要进行哪些方面的修改呀？是不是 label_str = '0 {:d} {:.6f} {:.6f} {:.6f} {:.6f}\n' 这句的第一个数据0要进行修改啊

[CaptainEven]

@ymRain 是的，第一列是class id， 具体的含义是: class_id, track_id, bbox_center_x, box_center_y, bbox_width, bbox_height, 祝顺利

[ymRain]

ok 谢谢

[CaptainEven]

@ymRain 方便分享一下你的数据集吗？方便的话我可以加到readme中去，让更多的人参与

[ymRain]

抱歉啊，我这个数据集没办法分享给你，这是个竞赛的数据集，我没有这个权限

[CaptainEven]

ok

[ymRain]

还有一个问题想请教您，如果我要检测十几个种类，训练时加载预训练模型应该加载mcmot_last_track_hrnet_18.pth这个模型吗？

[CaptainEven]

应该不行，heatmap数量不一致

[ymRain]

那您训练这个模型的时候是没有加载模型直接训练的？

[CaptainEven]

是的

[ymRain]

好的 谢谢

[ymRain]

大佬，麻烦您一下，运行track.py之后生成的txt文件每一列分别代表什么呀