GraphNN-Multi-Object-Tracking icon indicating copy to clipboard operation
GraphNN-Multi-Object-Tracking copied to clipboard

how can i use anthor reid model in this work

Open szh91 opened this issue 4 years ago • 1 comments

hi, thank you for the wonderful work and offer the code for us. If I train a separate ReID model, then I want to add this model in the code, how I can do this?

szh91 avatar Jan 07 '21 09:01 szh91

Hi, it should work pretty much out of the box by replacing the net variable here with your pretrained ReID model.

https://github.com/selflein/GraphNN-Multi-Object-Tracking/blob/74bf9698d9ea97ff9b94a7ff554f0f7ad15f27f3/src/data_utils/preprocessing.py#L201-L206

Note that this script uses PCA dimensionality reduction to reduce the feature size to 32 and is tuned for the ReID model used. So you may have to obtain a new PCA via https://github.com/selflein/GraphNN-Multi-Object-Tracking/blob/74bf9698d9ea97ff9b94a7ff554f0f7ad15f27f3/src/data_utils/preprocessing.py#L49 if feature sizes of your ReID method are not maching or remove the PCA entirely and add a MLP to the main model reducing the feature size.

selflein avatar Feb 03 '21 13:02 selflein