-
FairMOT-BDD100K是FairMOT的一个分支版本,用于在BDD100K数据集上进行MOT任务
资源介绍
FairMOT-BDD
这是FairMOT的一个分支,用于对BDD100K数据集进行MOT(多对象跟踪),也可以修改为其他自定义数据集。 你可以参考
主要贡献:修改原始代码以适应bdd100k数据集上的多类多对象跟踪训练和评估。
资料准备
首先,您需要将BDD100K数据集转换为MOT格式,例如文件夹“图像”和“带有ID的标签”。
对于培训MOT跟踪器:
python bdd2mot/bdd2mot.py --img_dir '/bdd_root/bdd100k/images/track' --label_dir '/bdd_root/bdd100k/labels-20/box-track' --save_path '/save_path/data/MOT'
对于训练探测器:
python bdd2mot/bdd2det.py --img_dir '/bdd_root/bdd100