Open GaoXinJian-USTC opened 3 years ago
Could you succeed to run the code if you only use 1 category in train_annots.csv and train_labels.csv?
Could you succeed to run the code if you only use 1 category in train_annots.csv and train_labels.csv?
Yes,but 12 categories can't
I am sorry that this version of CTracker may not support the multi-category tracking. Maybe you need to modify the code if you want to use CTracker to solve it.
I am sorry that this version of CTracker may not support the multi-category tracking. Maybe you need to modify the code if you want to use CTracker to solve it.
Can you give me some tips on how to modify it? I'm so confused ..XoX
For the detection part, you can refer to the source code of retinanet: https://github.com/yhenon/pytorch-retinanet For the tracking part, you need to rewrite test.py and modify the logic. For example, only the objects of the same category can be associated.
My labels.csv has 12 categories of num_classes. When it was raining, an error “The size of tensor a (512) must match the size of tensor b (12) at non-singleton dimension 1” occured.