Commit 25cd01e0 authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

bugfixed to process waymo data

parent 505e0ab3
...@@ -536,7 +536,7 @@ def create_waymo_gt_database( ...@@ -536,7 +536,7 @@ def create_waymo_gt_database(
workers=min(16, multiprocessing.cpu_count())): workers=min(16, multiprocessing.cpu_count())):
dataset = WaymoDataset( dataset = WaymoDataset(
dataset_cfg=dataset_cfg, class_names=class_names, root_path=data_path, dataset_cfg=dataset_cfg, class_names=class_names, root_path=data_path,
training=True, logger=common_utils.create_logger() training=False, logger=common_utils.create_logger()
) )
train_split = 'train' train_split = 'train'
train_filename = save_path / ('%s_infos_%s.pkl' % (processed_data_tag, train_split)) train_filename = save_path / ('%s_infos_%s.pkl' % (processed_data_tag, train_split))
......
...@@ -212,7 +212,7 @@ def process_single_sequence(sequence_file, save_path, sampled_interval, has_labe ...@@ -212,7 +212,7 @@ def process_single_sequence(sequence_file, save_path, sampled_interval, has_labe
return sequence_infos return sequence_infos
else: else:
sequence_infos_old = sequence_infos sequence_infos_old = sequence_infos
sequenec_infos = [] sequence_infos = []
for cnt, data in enumerate(dataset): for cnt, data in enumerate(dataset):
if cnt % sampled_interval != 0: if cnt % sampled_interval != 0:
......
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