/VectorNet

Pytorch implementation of paper "VectorNet: Encoding HD maps and Agent Dynamics from Vectorized Representation"

Primary LanguagePython

VectorNet

HitCount

Pytorch implementation of paper "VectorNet: Encoding HD maps and Agent Dynamics from Vectorized Representation".

目录结构:

.
├── README.md
├── data_process.py
├── docs
│   └── dataset.md
├── layers.py
├── model.py
├── run.py
└── save
    ├── features
    └── models

使用示例:

python3 run.py --root_dir ../forecasting_sample/data/ --epochs 50 --feature_path ./save/features/ --logging_steps 50 --train_batch_size=16 --enable_logging