detectRecog / PointTrack

PointTrack (ECCV2020 ORAL): Segment as Points for Efficient Online Multi-Object Tracking and Segmentation
Other
261 stars 47 forks source link

How to calculate category_embedding for my own dataset? #16

Closed ljjyxz123 closed 3 years ago

ljjyxz123 commented 3 years ago

In config.py, line 8, there is category_embedding definition: https://github.com/detectRecog/PointTrack/blob/master/config.py#L8 category_embedding = [[0.9479751586914062, 0.4561353325843811, 0.16707628965377808], [0.1,-0.1,0.1], [0.5455077290534973, -0.6193588972091675, -2.629554510116577], [-0.1,0.1,-0.1]]

How is this value calculated? If I want to train my own dataset, how to calculate the category_embedding for my own category?

Thanks! @detectRecog