search

ifzhang / FairMOT

[IJCV-2021] FairMOT: On the Fairness of Detection and Re-Identification in Multi-Object Tracking
MIT License
3.99k stars 935 forks source link

Extracting Re-id features #156

Closed CaucMark closed 4 years ago

CaucMark commented 4 years ago

Hi, thanks for your excellent work! Here's my question. I want to make some changes on Identity Embedding Branch, such as changing the way to extract the Re-id features. But I can't find the identity embedding branch in the code you provided(forgive me, I'm new in computer vision field). Would you please tell me where can I modify the code?

ifzhang commented 4 years ago

The channel of the embedding branch is set here: https://github.com/ifzhang/FairMOT/blob/ca78d3f01d1953c3d72ee8b01f24a5ddec2689c3/src/lib/opts.py#L217 The code of the network is here: https://github.com/ifzhang/FairMOT/blob/ca78d3f01d1953c3d72ee8b01f24a5ddec2689c3/src/lib/models/networks/pose_dla_dcn.py#L445

CaucMark commented 4 years ago

Thanks very much for your reply! It helps a lot~