396559551/social_lstm

Baseline implementation: https://github.com/ajvirgona/social-lstm-pytorch/tree/master/social_lstm
By Author: Anirudh Vemula
Date: 13th June 2017

Improvements:

1. Adjust test.py and train.py to accommodate vanilla lstm-> vlstm_train.py, vlstm_test.py
2. Adjust optimizer
3. reimplementing model.py and vlstm_model.py
4. Add visualization tools

Results
(from dataset 3)
Vanilla LSTM

Social LSTM

PLEASE SEE RESULTS FOLDER FOR TESTING RESULTS

TRAINING WITH SOCIAL LSTM run train.py run test.py

TRAINING WITH VANILLA LSTM run VLSTM_train.py run VLSTM_tets.py

MAKE VIDEO Animation.py

REPORT
Project.pdf