Fix api problem

mmdet3d/datasets/kitti_dataset.py

@@ -115,7 +115,8 @@ class KittiDataset(Custom3DDataset):
             gt_bboxes_3d=gt_bboxes_3d,
             gt_labels_3d=gt_labels_3d,
             bboxes=gt_bboxes,
-            labels=gt_labels)
+            labels=gt_labels,
+            gt_names=gt_names)
         return anns_results
 
     def drop_arrays_by_name(self, gt_names, used_classes):

tools/data_converter/create_gt_database.py

@@ -144,12 +144,12 @@ def create_groundtruth_database(dataset_class_name,
     print(f'Create GT Database of {dataset_class_name}')
     dataset_cfg = dict(
         type=dataset_class_name,
-        root_path=data_path,
+        data_root=data_path,
         ann_file=info_path,
     )
     if dataset_class_name == 'KittiDataset':
         dataset_cfg.update(
-            training=True,
+            test_mode=False,
             split='training',
             modality=dict(
                 use_lidar=True,
@@ -178,11 +178,11 @@ def create_groundtruth_database(dataset_class_name,
 
     group_counter = 0
     for j in track_iter_progress(list(range(len(dataset)))):
-        image_idx = j
-        annos = dataset.get_sensor_data(j)
+        annos = dataset.get_data_info(j)
         image_idx = annos['sample_idx']
-        points = annos['points']
-        gt_boxes_3d = annos['gt_bboxes_3d']
+        points = np.fromfile(
+            annos['pts_file_name'], dtype=np.float32).reshape(-1, 4)
+        gt_boxes_3d = annos['ann_info']['gt_bboxes_3d']
         names = annos['gt_names']
         group_dict = dict()
         group_ids = np.full([gt_boxes_3d.shape[0]], -1, dtype=np.int64)

tools/data_converter/kitti_converter.py

@@ -92,9 +92,8 @@ def create_kitti_info_file(data_path,
         relative_path (bool): Whether to use relative path.
     """
     imageset_folder = Path(data_path) / 'ImageSets'
-    train_img_ids = _read_imageset_file(
-        str(imageset_folder / 'train_6014.txt'))
-    val_img_ids = _read_imageset_file(str(imageset_folder / 'val_1467.txt'))
+    train_img_ids = _read_imageset_file(str(imageset_folder / 'train.txt'))
+    val_img_ids = _read_imageset_file(str(imageset_folder / 'val.txt'))
     test_img_ids = _read_imageset_file(str(imageset_folder / 'test.txt'))
 
     print('Generate info. this may take several minutes.')