GithubHelp home page GithubHelp logo

haoran-song / pip-planning-informed-prediction Goto Github PK

View Code? Open in Web Editor NEW
170.0 9.0 45.0 223 KB

(ECCV 2020) PiP: Planning-informed Trajectory Prediction for Autonomous Driving

Home Page: http://song-haoran.com/planning-informed-prediction/

Python 65.08% MATLAB 34.44% Shell 0.48%
deep-learning motion-forecasting trajectory-prediction trajectory-planning autonomous-driving eccv2020

pip-planning-informed-prediction's Issues

About NGSIM dataset

I'm sorry to bother you,It seems that the data I downloaded from the NGSIM is not complete. Could you send me a copy of your original data:us101-0750am-0805am-0820am-0835am.txt , i80-0400-0415.txt and i80-0500-0515-0530.txt.
Thank you so much!

How to get the ego vehicle's future planning trajectory without ground truth

Nice work. I have a question about the ego vehicle's future planning. In this work, it has been used as input. But for testing and evaluation, how can we get the ego vehicle's future planning trajectory if we do not have the ground truth? It seems that the ego vehicle's future planning trajectory is generated from the ground truth in the code.
Line 325 in data.py:

` def getPlanFuture(self, dsId, planId, refVehId, t):
# Traj of the reference veh
refColIndex = np.where(self.Tracks[dsId - 1][refVehId - 1][0, :] == t)[0][0]
refPos = self.Tracks[dsId - 1][refVehId - 1][1:3, refColIndex].transpose()
# Traj of the planned veh
planColIndex = np.where(self.Tracks[dsId - 1][planId - 1][0, :] == t)[0][0]
stpt = planColIndex
enpt = planColIndex + self.t_f + 1
planGroundTrue = self.Tracks[dsId - 1][planId - 1][1:3, stpt:enpt:self.d_s].transpose()
planFut = planGroundTrue.copy()
# Fitting the downsampled waypoints as the planned trajectory in testing and evaluation.
if self.fit_plan_traj:
wayPoint = np.arange(0, self.t_f + self.d_s, self.d_s)
wayPoint_to_fit = np.arange(0, self.t_f + 1, self.d_s * self.further_ds_plan)
planFut_to_fit = planFut[::self.further_ds_plan, ]
laterParam = fitting_traj_by_qs(wayPoint_to_fit, planFut_to_fit[:, 0])
longiParam = fitting_traj_by_qs(wayPoint_to_fit, planFut_to_fit[:, 1])
planFut[:, 0] = quintic_spline(wayPoint, *laterParam)
planFut[:, 1] = quintic_spline(wayPoint, *longiParam)'

    revPlanFut = np.flip(planFut[1:,] - refPos, axis=0).copy()
    return revPlanFut `

Training time

Thanks for your awesome work!
I'd like to ask about training time?
I found it takes me about 5 hours for one epoch.
Is it right?
Thanks!

数据预处理

您好,我在读您这篇文章代码的时候发现在处理目标车辆的历史轨迹数据与邻居车辆的历史轨迹数据时有 wholePeriod 参数,在处理目标车辆历史轨迹时wholePeriod参数置0,即以最后观察时刻为原点,在处理邻居车辆历史轨迹时wholePeriod参数置1,即用邻居车辆每一时刻的坐标减目标车辆对应时刻的坐标,这会使得邻居车辆与目标车辆不在同一坐标系中。所以为什么要在处理邻居车辆历史轨迹时将wholePeriod参数置1呢?

memory

Dear author, which of the default parameters should I adjust when I don't have enough memory?

可视化预测结果

您好:
非常棒的工作并且开源了代码。
想问下是否可开源预测结果可视化的代码?非常感谢。

Can this program predict the situation of intersections?

Hi,
Thank you very much for your excellent work! If I want to apply your work to intersections, what changes do I need to make? Can your achievements be directly applied to the intersections? And I want to predict the traffic situation in the Interaction dataset. What do I need to do? (I'm a beginner, and I don't know the logic at the bottom of the network. Please tell me more, thank you.)
Looking forward to your reply.

数据集下载解压失败

你好,我在你给其他人回复的链接里下载三个数据集 但都是解压失败 是文件本身上传出错了吗 我已经试过好多次了,三个数据集都不行。。。。

报错问题

您好,我有一个问题,当我关闭planning模块时,evaluate会报错:RuntimeError: Error(s) in loading state_dict for pipNet:
Unexpected key(s) in state_dict: "plan_lstm.weight_ih_l0", "plan_lstm.weight_hh_l0", "plan_lstm.bias_ih_l0", "plan_lstm.bias_hh_l0", "plan_conv_social.0.weight", "plan_conv_soc
ial.0.bias", "plan_conv_social.3.weight", "plan_conv_social.3.bias".
我不知道如何解决,希望您百忙之中可以解答我的问题

Trained models

Hello, thank you very much for the source code for vehicle trajectory prediction. When I unzip the downloaded trained model ,there will always be errors. It has been solved for a long time without success. Can you send me a copy to my email “[email protected]”? thank you.

About Preprocessing HighD Dataset

Thank you for providing source code of your brilliant work on vehicle trajectory prediction. I have 2 questions about HighD dataset using in your work:

  1. In the file 'preprocess_highD.m', the length of past trajectory is 30 frames and that of future trajectory is 50 frames, which is the same to your implementation in 'preprocess_ngsim.m'. However, as far as I know, the sampling rate of HighD dataset is 25Hz and is different from the sampling rate of NGSIM dataset, which is 10Hz. Could you please share the details about dealing with such problem?
  2. It would be very kind of you if the source code on plotting the figures in your paper will be shared.

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.