Skip to content

Commit 378593a

Browse files
authored
Add pandaset support (#396)
* Add basic support for Pandaset dataset * Add gt_sampling support for pandaset * Enable training on different classes than raw dataset * Save pandaset output in the same format as annotations * Add dataset config file for pandaset * Make pandaset dataset compatible with test.py - add 'evaluation' method - add 'name' key to predictions dict * Enable selecting lidar device for pandaset
1 parent e94c91d commit 378593a

File tree

3 files changed

+599
-1
lines changed

3 files changed

+599
-1
lines changed

pcdet/datasets/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,14 @@
88
from .kitti.kitti_dataset import KittiDataset
99
from .nuscenes.nuscenes_dataset import NuScenesDataset
1010
from .waymo.waymo_dataset import WaymoDataset
11+
from .pandaset.pandaset_dataset import PandasetDataset
1112

1213
__all__ = {
1314
'DatasetTemplate': DatasetTemplate,
1415
'KittiDataset': KittiDataset,
1516
'NuScenesDataset': NuScenesDataset,
16-
'WaymoDataset': WaymoDataset
17+
'WaymoDataset': WaymoDataset,
18+
'PandasetDataset': PandasetDataset
1719
}
1820

1921

0 commit comments

Comments
 (0)