-
EDA_GNN:一个用于在线多目标跟踪的端到端图神经网络数据关联框架
资源介绍
在线多目标跟踪的基于图神经的端到端数据关联框架
PyTorch实现与暹罗网络和图神经网络相结合,用于在线多对象跟踪。
数据集可在[ ]
可以在[ ]中找到相应的论文。
怎么跑
使用python main.py从头开始训练模型。 培训设置位于config.yml 。 使用python tracking.py跟踪测试视频,同时您需要提供前五个帧的检测到的对象和跟踪结果。 跟踪设置在setting/ 。
要求
Python 2.7.12
的numpy 1.11.0
scipy 1.1.0
火炬视觉0.2.1
opencv_python 3.3.0.10
easydict 1.7
火炬0.4.1
枕头6.2.0
PyYAML 5.1