Commit fc258622 authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

small update of WOD

parent ae04946d
...@@ -87,8 +87,8 @@ pip3 install waymo-open-dataset-tf-2-0-0==1.2.0 --user ...@@ -87,8 +87,8 @@ pip3 install waymo-open-dataset-tf-2-0-0==1.2.0 --user
* Extract point cloud data from tfrecord and generate data infos by running the following command (it takes several hours): * Extract point cloud data from tfrecord and generate data infos by running the following command (it takes several hours):
```python ```python
python -m pcdet.datasets.waymo.waymo_dataset --func create_waymo_infos \ - python -m pcdet.datasets.waymo.waymo_dataset --func create_waymo_infos \
-cfg_file tools/cfgs/dataset_configs/waymo_dataset.yaml --cfg_file tools/cfgs/dataset_configs/waymo_dataset.yaml
``` ```
Note that you do not need to install `waymo-open-dataset` if you have already processed the data before and do not need to evaluate with official Waymo Metrics. Note that you do not need to install `waymo-open-dataset` if you have already processed the data before and do not need to evaluate with official Waymo Metrics.
......
...@@ -90,7 +90,7 @@ class WaymoDataset(DatasetTemplate): ...@@ -90,7 +90,7 @@ class WaymoDataset(DatasetTemplate):
for sequence_file in self.sample_sequence_list for sequence_file in self.sample_sequence_list
] ]
process_single_sequence(sample_sequence_file_list[0]) # process_single_sequence(sample_sequence_file_list[0])
with futures.ThreadPoolExecutor(num_workers) as executor: with futures.ThreadPoolExecutor(num_workers) as executor:
sequence_infos = executor.map(process_single_sequence, sample_sequence_file_list) sequence_infos = executor.map(process_single_sequence, sample_sequence_file_list)
sequence_infos = list(sequence_infos) sequence_infos = list(sequence_infos)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment