Merge pull request #121 from zyc10ud/maptrv2

update v2 visualization scripts and doc
hustvl · Oct 6, 2023 · e03f097 · e03f097
2 parents ebcc459 + 5e6f111
commit e03f097
Show file tree

Hide file tree

Showing 3 changed files with 907 additions and 1 deletion.
diff --git a/docs/visualization.md b/docs/visualization.md
@@ -7,7 +7,11 @@ We provide all the visualization scripts under `/path/to/MapTR/tools/maptr/`
 ```shell
 cd /path/to/MapTR/
 export PYTHONPATH="/path/to/MapTR/"
-python tools/maptr/vis_pred.py /path/to/experiment/config /path/to/experiment/ckpt
+# visualize nuscenes dataset
+python tools/maptrv2/nusc_vis_pred.py /path/to/experiment/config /path/to/experiment/ckpt
+
+#visualize argoverse2 dataset
+python tools/maptrv2/av2_vis_pred.py /path/to/experiment/config /path/to/experiment/ckpt
 ```
 **Notes**: