timmeinhardt / trackformer

Implementation of "TrackFormer: Multi-Object Tracking with Transformers”. [Conference on Computer Vision and Pattern Recognition (CVPR), 2022]
https://arxiv.org/abs/2101.02702
Apache License 2.0
487 stars 113 forks source link

Custom dataset: class_error: 0 #110

Open nikeshdevkota opened 9 months ago

nikeshdevkota commented 9 months ago

Hello,

I'm trying to train a custom dataset that has only one target per image. Currently, I am getting a class error of 0. Is this correct? I was not sure about how to set "num_queries" parameter for a single class (and single object) per image, so I have set it to 10 randomly for now.

I am using the command python src/train.py with infrared deformable multi_frame tracking output_dir=models/infrared_deformable_multi_frame

I have already set the custom data load function similar to mot17 and set my class_id as 1.Since the tracking_eval during training needs a custom dataloader or similar format to original mot17, I have set it as False for now.

Zero Class Error Custom Dataset Train_Tracking