/Safe_Goodbye

Prediction of Passengers' Intention to Get off the City Buses

Primary LanguagePython

Safe Goodbye

TEAM: Arrival(Jungyeon Lee/Chaeeun Kim/Yookyung Jeong)

If you click the image below, you can see the presentation video on Youtube.

Watch the video

Dataset

  • Removed List
>>> image problem
...\ele\[ele]attend_270C
...\etc\[etc]attend_154C
...\hospital\[hospital]attend_392C
...\hospital\[hospital]leave_251C
>>> label problem
...\apt\[apt]attend_1275C
...\apt\[apt]attend_941C
  • Target(demonstration result) from Validation set
reselecting

type 1
[hospital]attend_067C

type 2
[mid_high]leave_354C

Train Valid
per Scene
per Frame

Modeling

  • 하차 전 약 35프레임에서의 각 사람의 skeleton data를 time series data로 만들어서 하차할 것인지 하차하지 않을 것인지 Intention Prediction

Train Valid
Loss
Accuracy

Codes

  • data_analysis.py: 1
  • make_dataset.py: 2
  • dataset.py: 3
  • simple_dataset_keras.py: 4 simple lstm model로 training
  • train.py: 5
  • generate_video.py: 6
  • tf2_multipose.py: todo

Reference

Baseline

OD+Tracking

Pose Estimation