how to train and predict on own custom dataset?

Khrylx / AgentFormer

[ICCV 2021] Official PyTorch Implementation of "AgentFormer: Agent-Aware Transformers for Socio-Temporal Multi-Agent Forecasting".

https://www.ye-yuan.com/agentformer/

MIT License

257 stars 64 forks source link

how to train and predict on own custom dataset? #39

Open poincarelee opened 2 months ago

poincarelee commented 2 months ago

Hi , your have done a really wonderful work. I am working on your code and I want to know how to train and predict on own data. For example, I have photoed some pedestrain walking videos, how can I train on my own data and predict the pedestrian trajectory?