search

jzuern / autograph-pub

Repository for the publication "AutoGraph: Predicting Lane Graphs from Traffic"
http://autograph.cs.uni-freiburg.de/
6 stars 2 forks source link

For Running Instruction #1

Open qiaozhijian opened 3 months ago

qiaozhijian commented 3 months ago

Hi, thanks for your nice work. I have downloaded the dataset, and its folders are like this

MY_PATH/urbanlanegraph-dataset-pub-v1.1
├── austin
│   ├── austin_centerlines.png
│   ├── austin_direction.png
│   ├── austin_drivable.png
│   ├── austin_intersections.png
│   ├── austin.png
│   ├── austin-tracklets.png
│   ├── austin_viz.png
│   ├── successor-lgp
│   └── tiles
├── detroit
├── miami
├── paloalto
├── pittsburgh
├── version.txt
└── washington

I see there are at least three parameters to be set. They are dataroot, dataroot_urbanlanegraph and dataset_name.

How to set these three variables? Could you please give an example to run the train and evaluation file? BTW, what's your GPU type and number?

qiaozhijian commented 3 months ago

What's more, there seems to be missing packages when I run train_regressor_pos_query.py:

from urbanlanegraph_evaluator.evaluator import GraphEvaluator
from urbanlanegraph_evaluator.utils import adjust_node_positions