RayneSun / VTPLSTM Public

Notifications You must be signed in to change notification settings
Fork 1
Star 32

Inspired by the performence of Social LSTM. I created a model that could predict vehicle's trajectory in 5s. The model uses LSTM as center. I trained it at NGSIM. Still fix it...

32 stars 1 fork Branches Tags Activity

Star

Notifications

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
README.md		README.md
data_generator.py		data_generator.py
data_loader.py		data_loader.py
model.py		model.py
parameters.py		parameters.py
train.py		train.py
utils.py		utils.py
visualize.py		visualize.py

Repository files navigation

VTPLSTM

Inspired by the performence of Social LSTM. I created a model that could predict vehicle's trajectory in 5s. The model uses LSTM as center. I trained it at NGSIM. Still fix it...

1.DownLoad NGSIM dataset(.csv) 2.Fix parameters in parameters.py 3.Extract dataset by data_generator.py 4.Train it (fix model by parameters.py) 5.visualize by visualdl and visuallize

I will complete this project in the next month.

About

Inspired by the performence of Social LSTM. I created a model that could predict vehicle's trajectory in 5s. The model uses LSTM as center. I trained it at NGSIM. Still fix it...

Readme

Activity

32 stars

1 watching

1 fork

Report repository