GraphNN-Multi-Object-Tracking
GraphNN-Multi-Object-Tracking copied to clipboard
how can i use anthor reid model in this work
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?
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.