GithubHelp home page GithubHelp logo

mengmengliu1998 / gatraj Goto Github PK

View Code? Open in Web Editor NEW
49.0 5.0 7.0 24.13 MB

[ISPRS 2023]Official PyTorch Implementation of "GATraj: A Graph- and Attention-based Multi-Agent Trajectory Prediction Model"

License: MIT License

Python 100.00%
autonomous-driving graph-convolutional-networks multi-agent-forecasting pytorch self-attention trajectory-prediction

gatraj's People

Contributors

haohao11 avatar mengmengliu1998 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar

gatraj's Issues

visualization

Thank you for your excellent work. Hello, I have some problems with visualization. If possible, can you share the source code?

Question about multihead_proj_global

Thank for your outstanding work!

  1. global_embed = self.multihead_proj_global(global_embed).view(-1, self.num_modes, self.hidden_size) # [N, F, D]

What is the purpose of this function? Why design this function?

  1. Here

mdn_out = self.Laplacian_Decoder.forward(self.x_encoded_dense, self.hidden_state_global, cn_global, epoch)

the parameters of this function are time tensor x_encoded_dense and spatial tensor hidden_state_global

but in

def forward(self, x_encode: torch.Tensor, hidden_state, cn) -> Tuple[torch.Tensor, torch.Tensor]:

the time tensor's name change to global_embed and the spatial tensor’s name change to local_embed, Is this correct?

Thank you!

dataset

Thank you for your excellent work. Can you please provide the nuScenes dataset?

the option ”--input-offset“

Thank you for your outstanding work !

There is an option called --input_offset in the program.

When this option is set to True, the train_x is the difference between batch_norm_gt , batch_norm_gt is the difference between absolute position. why do we do this?

train_x = batch_norm_gt[1:self.args.obs_length, :, :] - batch_norm_gt[:self.args.obs_length-1, :, :] #[H, N, 2]

code on nuScenes

Hi, @mengmengliu1998 I have had the pleasure of reading your paper, I feel it is very well written, now I want to study the code of the paper, I see you have provided the code on ETH/UCY dataset, I would like to ask if you are convenient to provide the code on nuScenes dataset, if it is convenient can you send it to my email, [email protected],

nuScene dataset

Hello, I have recently read your article and your proposed framework is excellent. I would like to further understand how the model is applied to the nuScene dataset? I would like to ask if you are convenient to provide the code on nuScenes dataset. Thank you! This is my email: [email protected]

question about mode k and likelihood π

hello,I have a question regarding the implementation of mode k and the likelihood π in your algorithm.

I would like to know whether mode k is applied to individual agents independently, or is it used to describe all agents within the entire scene? In othjer words, are the likelihoods π associated with modes k the same for all agents, or does each agent have distinct likelihoods π for their respective modes k?

looking forward to your reply!! thank you!

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.