Merge pull request #6 from Mrmoore98/main

init commit

Author: Mrmoore98

README.md

@@ -1,10 +1,11 @@
 # VectorMapNet_code
 **VectorMapNet: End-to-end Vectorized HD Map Learning**
 
-This is the offical codebase of VectorMapNet
+This is the official codebase of VectorMapNet
 
 
-[Yicheng Liu](https://scholar.google.com/citations?user=vRmsgQUAAAAJ&hl=zh-CN), [Yue Wang](https://people.csail.mit.edu/yuewang/), [Yilun Wang](https://scholar.google.com.hk/citations?user=nUyTDosAAAAJ&hl=en/), [Hang Zhao](http://people.csail.mit.edu/hangzhao/)
+[Yicheng Liu](https://scholar.google.com/citations?user=vRmsgQUAAAAJ&hl=zh-CN), Yuantian Yuan, [Yue Wang](https://people.csail.mit.edu/yuewang/), [Yilun Wang](https://scholar.google.com.hk/citations?user=nUyTDosAAAAJ&hl=en/), [Hang Zhao](http://people.csail.mit.edu/hangzhao/)
+
 
 **[[Paper](https://arxiv.org/pdf/2206.08920.pdf)] [[Project Page](https://tsinghua-mars-lab.github.io/vectormapnet/)]**
 
@@ -15,7 +16,7 @@ Autonomous driving systems require a good understanding of surrounding environme
 Please file an [issue](https://github.com/Tsinghua-MARS-Lab/vecmapnet/issues) or send an email to [Yicheng]([email protected]).
 
 
-### Citation
+## Bibtex
 If you found this paper or codebase useful, please cite our paper:
 ```
 @article{liu2022vectormapnet,
@@ -25,3 +26,63 @@ If you found this paper or codebase useful, please cite our paper:
     year={2022}
     }
 ```
+
+
+# Run VectorMapNet
+
+## Note
+We are still working on refining this repo. Currently, we only release the inference code of VectorMapNet.
+
+The training pipeline will be released in next week. 
+
+## 0. Environment
+
+Set up environment by following this [script](env.md)
+
+## 1. Prepare your dataset
+
+Store your data with following structure:
+
+```
+    root
+        |--datasets
+            |--nuScenes
+            |--Argoverse2(optional)
+
+```
+
+### 1.1 Generate annotation files
+
+#### Preprocess nuScenes
+
+```
+python tools/data_converter/nuscenes_converter.py --data-root your/dataset/nuScenes/
+```
+
+## 2. Evaluate VectorMapNet
+
+### Download Checkpoint
+| Method       | Modality    | Config | Checkpoint |
+|--------------|-------------|--------|------------|
+| VectorMapNet | Camera only | [config](configs/vectormapnet.py) | [model link](https://drive.google.com/file/d/1ccrlZ2HrFfpBB27kC9DkwCYWlTUpgmin/view?usp=sharing)      |
+
+### Do Evaluation
+
+In single GPU
+```
+python tools/test.py configs/vectormapnet.py /path/to/ckpt --eval name
+```
+
+For multi GPUs
+```
+bash tools/dist_test.sh configs/vectormapnet.py /path/to/ckpt $num_gpu --eval name
+```
+
+
+### Expected Results
+
+| $AP_{ped}$   | $AP_{divider}$ | $AP_{boundary}$ | mAP   |
+|--------------|----------------|-----------------|-------|
+| 39.8 | 47.7    | 38.8          | 42.1 |
+
+

configs/_base_/default_runtime.py

@@ -0,0 +1,18 @@
+checkpoint_config = dict(interval=1)
+# yapf:disable push
+# By default we use textlogger hook and tensorboard
+# For more loggers see
+# https://mmcv.readthedocs.io/en/latest/api.html#mmcv.runner.LoggerHook
+log_config = dict(
+    interval=50,
+    hooks=[
+        dict(type='TextLoggerHook'),
+        dict(type='TensorboardLoggerHook')
+    ])
+# yapf:enable
+dist_params = dict(backend='nccl')
+log_level = 'INFO'
+work_dir = None
+load_from = None
+resume_from = None
+workflow = [('train', 1)]