Releases: selflein/GraphNN-Multi-Object-Tracking
Releases · selflein/GraphNN-Multi-Object-Tracking
Update to PyTorch and PyG LTS releases
- Update PyTorch to 1.8.2 LTS
- Update pytorch-geometric to 1.7.2
- Rewrite message passing to use
aggregate(...)instead of overwriting
propagate - Remove PCA dimensionality reduction and use learnable MLP to reduce
node feature dimensionalityd
Pre-trained models
Add pre-trained models
pca.sklearnis used for dimensionality reduction of visual features in pre-processing scriptfaster_rcnn_fpn.modelis the object detection modelgraph_nn_bce_30.pthis the trained GNN model for tracking