Stochastic Scene-Aware Motion Prediction

Stochastic Scene-Aware Motion Prediction

[Project Page]
[Paper]

Description

This repository contains the training code for MotionNet and GoalNet of SAMP.

Installation

To install the necessary dependencies run the following command:

    pip install -r requirements.txt

The code has been tested with Python 3.8.10, CUDA 10.0, CuDNN 7.5 and PyTorch 1.7.1 on Ubuntu 20.04.

Training Data

The training data for MotionNet and GoalNet could be found in the website downloads. Or could be extracted
from the Unity runtime code.

Update

To finish reading, please visit source site