/Multi-Class-Social-STGCNN

Building on the work of Social STGCNN we introduce Multi-Class Social STGCNN to include class information into trajectory prediction to improve the accuracy.

Primary LanguagePython

Stargazers