Skip to content
Social LSTM implementation in PyTorch
Python Shell
Branch: master
Clone or download
Baran Nama
Baran Nama Merge pull request #13 from mirsking/master
fix error with torch 1.1
Latest commit 9fa3007 May 14, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data first commit Jun 21, 2018
README.md
generator.py error fix Jun 22, 2018
grid.py error fix Jun 22, 2018
helper.py fix error with torch 1.1 May 14, 2019
hyperparameter.py error fix Jun 22, 2018
make_directories.sh first commit Jun 21, 2018
model.py cuda error fix 1 Jun 21, 2018
olstm_model.py comment addition Jun 21, 2018
olstm_train.py error fix Jun 22, 2018
test.py error fix Jun 22, 2018
train.py fix error with torch 1.1 May 14, 2019
utils.py error fix Jun 22, 2018
validation.py error fix Jun 22, 2018
visualize.py error fix Jun 22, 2018
vlstm_model.py comment addition Jun 21, 2018
vlstm_train.py error fix Jun 22, 2018

README.md

Social LSTM implementation in PyTorch

Project details

Semester project of Master of Computer Science in EPFL
Student name : Baran Nama
Advisor: Alexandre Alahi
Presentation : https://drive.google.com/file/d/1biC23s1tbsyDETKKBW8PFXWYyyhNEAuI/view?usp=sharing

Implementation details

Baseline implementation: https://github.com/vvanirudh/social-lstm-pytorch
Paper: http://cvgl.stanford.edu/papers/CVPR16_Social_LSTM.pdf
Made improvements: Please see attached presentation

Documentation

  • generator.py : Python script for generating artifical datasets
  • helper.py: Python script includes various helper methods
  • hyperparameter.py: Pyton script for random best parameter selection for a model
  • make_directories.sh: Bash script for creation of file structure
  • model.py: Python file includes Social LSTM model definition
  • olstm_model.py: Python file includes Occupancy LSTM model definition
  • olstm_train.py: Python script for training Occupancy LSTM model
  • test.py: Python script for model testing and getting output txt file for submission
  • train.py: Python script for training Social LSTM model
  • utils.py: Python script for handling input train/test/validation data and batching it
  • validation.py: Python script for externally evaluate a trained model by getting validation error
  • visualize.py: Python script for visualizing predicted trajectories during train/test/validation sessions
  • vlstm_model.py: Python file includes Vanilla LSTM model definition
  • vlstm_train: Python script for training Vanilla LSTM model

How to deploy

  1. Fork the repository
  2. Start train a model >>> python train/olstm_train/vlstm.train.py - -[Parameter set]
  3. If necesarry file structure is not exist (which is the initial situation), train script will run make_directories.sh and this command will automatically create file structure
  4. Enjoy!

Results

Model name Avarage error Final error Mean error
Social LSTM 1.3865 2.098 0.675
Occupancy LSTM 2.1105 3.12 1.101
Vanilla LSTM 2.107 3.114 1.1

Reference: http://trajnet.stanford.edu/result.php?cid=1

You can’t perform that action at this time.