How to track only one category in coco80 datasets without training myself?

[dongziqi001]

This code is great!! I have ran the demo to track other objects(cars) in videos,which is one of the types of category in coco. but I have two questions. 1.How can I get the video tracking results? 2.How to track only one category in coco80 datasets without training myself? Thanks!!!

[xingyizhou]

1. When you call ret = detector.run(img)['results'], ret will be a list dict: [{'bbox': [x1, y1, x2, y2], 'tracking_id': id, ...}]. You can use this to render your output video.
2. In ret above, simply filter out the item with 'category_id' != 2. The category_id list is here.

[dongziqi001]

Thank you for your kindly help. 1.After I got list dict: [{'bbox': [x1, y1, x2, y2], 'tracking_id': id, ...}],How can I convert it as output.mp4?

2. I modified "results" in detector.py and filtered car category. Thanks! @xingyizhou

[Veronica1997]

@dongziqi001 I have the same question with you , Have you solved the problem of convert it as output.mp4? Thanks!

[ZhinanCai]

Thank you for your kindly help. 1.After I got list dict: [{'bbox': [x1, y1, x2, y2], 'tracking_id': id, ...}],How can I convert it as output.mp4?

1. I modified "results" in detector.py and filtered car category. Thanks! @xingyizhou

hi, I got the same problem,I wonder how do you modified "results" in detector.py, would you share with me please?

[dodeokwangyooo]

Thank you for your kindly help. 1.After I got list dict: [{'bbox': [x1, y1, x2, y2], 'tracking_id': id, ...}],How can I convert it as output.mp4?

1. I modified "results" in detector.py and filtered car category. Thanks! @xingyizhou

hi, I got the same problem,I wonder how do you modified "results" in detector.py, would you share with me please?

have u solved this problem? can u tell me how to get it