huangfu/scenarionet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix a bug in Waymo conversion: GPU should be disable (#64)

Browse Source 
* Update waymo.rst

* Update waymo.rst

* allow generate all data

* update readme

* update

* better logging info

* more info

* up

* fix

* add note on GPU

* better log

* format
This commit is contained in:
Zhenghao Peng
2024-02-20 13:28:09 -08:00
committed by GitHub
parent 06c3aee0e2
commit 6cda061ed8
8 changed files with 47 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
documentation/waymo.rst
View File

@@ -35,7 +35,7 @@ First of all, we have to install tensorflow and Protobuf::
conda install protobuf==3.20
.. note::
You may fail to install ``protobuf`` if using ``pip install protobuf==3.20``.
You may fail to install ``protobuf`` if using ``pip install protobuf==3.20``. If so, install via ``conda install protobuf=3.20``.
2. Download TFRecord
@@ -79,12 +79,18 @@ The downloaded data should be stored in a directory like this::
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Run the following command to extract scenarios in any directory containing ``tfrecord``.
Here we take converting raw data in ``training_20s`` as an example::
python -m scenarionet.convert_waymo -d /path/to/your/database --raw_data_path ./waymo/training_20s --num_files=1000
python -m scenarionet.convert_waymo -d /path/to/your/database --raw_data_path ./waymo/training_20s --num_workers 64
Now all converted scenarios will be placed at ``/path/to/your/database`` and are ready to be used in your work.
.. note::
When running the conversion, please double check whether GPU is being used. This converter should NOT use GPU.
We have disable GPU usage by ``os.environ["CUDA_VISIBLE_DEVICES"] = ""``.
Known Issues: Waymo
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
N/A