(Fork of) Interaction Modeling with Multiplex Attention

Authors: Fan-Yun Sun, Isaac Kauvar, Ruohan Zhang, Jiachen Li, Mykel Kochenderfer, Jiajun Wu, Nick Haber

This repository contains the abbrivaited version of the above paper (arxiv, openreview). The original repo can be found here.

Environment Setup

1. Install Python-RVO2 library
2. Install socialforce library
3. Install necessary packages with pip

pip install -r requirements.txt

Data Setup

Download the preprocessed dataset here (or run gdown 'https://drive.google.com/drive/folders/14bxUhp2K4BZYFk9d3wQkWj9BLbY36VAy?usp=sharing') and place it under datasets.

Run code

run_bball.py
run_spring.py