Skip to content

Releases: selflein/GraphNN-Multi-Object-Tracking

Update to PyTorch and PyG LTS releases

05 Dec 17:57
7207e74

Choose a tag to compare

  • 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

13 Dec 19:43

Choose a tag to compare

Add pre-trained models

  • pca.sklearn is used for dimensionality reduction of visual features in pre-processing script
  • faster_rcnn_fpn.model is the object detection model
  • graph_nn_bce_30.pth is the trained GNN model for tracking