4 个月前

Social GAN:基于生成对抗网络的社会可接受轨迹预测

Social GAN:基于生成对抗网络的社会可接受轨迹预测

摘要

理解人类运动行为对于自主移动平台(如自动驾驶汽车和社会机器人)在以人类为中心的环境中导航至关重要。这一问题具有挑战性,因为人类运动本质上是多模态的:给定一段人类运动路径的历史记录,未来人们可能采取的行动方式有多种社会上合理的可能性。我们通过结合序列预测和生成对抗网络的工具来解决这一问题:一个循环序列到序列模型观察运动历史并预测未来行为,利用一种新颖的池化机制来聚合人群中的信息。我们通过对抗性训练与一个循环判别器竞争,以预测社会上合理的行为,并通过引入一种新颖的多样性损失函数来鼓励多样性的预测。通过在多个数据集上的实验,我们证明了我们的方法在准确性、多样性、碰撞避免和计算复杂度方面优于先前的工作。

代码仓库

mirkozaff/aa-sgan
pytorch
GitHub 中提及
cmubig/SPEC
pytorch
GitHub 中提及
m-hasan-n/pooling
pytorch
GitHub 中提及
romi514/TLSGAN
pytorch
GitHub 中提及
rohanchandra30/TrackNPred
pytorch
GitHub 中提及
amiryanj/socialways
pytorch
GitHub 中提及
agrimgupta92/sgan
官方
pytorch
GitHub 中提及
huang-xx/STGAT
pytorch
GitHub 中提及

基准测试

基准方法指标
trajectory-prediction-on-ethSocial-GAN
Avg AMD/AMV 8/12: 1.42
trajectory-prediction-on-stanford-droneSocial GAN
ADE (8/12) @K=5: 27.25
ADE-8/12 @K = 20: 27.23
FDE(8/12) @K=5: 41.44
FDE-8/12 @K= 20: 41.44

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
Social GAN:基于生成对抗网络的社会可接受轨迹预测 | 论文 | HyperAI超神经