论文阅读:Social LSTM: Human Trajectory Prediction in Crowded Spaces

时间:2024-04-12 16:22:32

一、关于social-lstm

论文阅读:Social LSTM: Human Trajectory Prediction in Crowded Spaces
1.文中提到social force model指的是一种吸引力(attractive)和排斥力(repulsive)的行人运动模型
2.如上图所示,为4个人分别构建一个lstm模型,然后在他们之间加入一个social池化层,这样搭建出来的网络就是social-lstm。
3.social池化层主要是用来共享隐藏状态hidden-states的。

二、实验

github上面有一个这个项目用的是pytorch:项目地址
然后自己下载下来跑了一下,模型跑的没问题,但是可视化的时候遇到了一些问题,无法接受参数,感觉可能是系统的原因,因为作者是在linux下跑的。然后自己改了一下,解决了可视化图片的问题,但具体的意思还不大理解。
其中转成视频也有点问题,按照issues里面做了,也没有解决。
接下来就准备在把可视化为视频解决之后,重新写一下这个项目,只用他的social-lstm,然后跑自己的模型试试。