[Registry] How to properly load the registry ?

[IdrissGo]

Using standard kitti dataset with nothing changed. I wanted to do some statistics on the data but I can't build the dataset. It seems that the registry does not register any of the transforms.

I try to import mmengine, mcv and mmdet3d but it doesn't fix the issue.

How to make sure that all the modules/datasets/transforms are correctly registered in the registry ?

Code :

import numpy as np 
from mmdet3d.registry import DATASETS
import mmcv
import mmengine
import mmdet3d



if __name__ == '__main__':
    dataset_class_name = 'KittiDataset'
    data_path = "./data/kitti"
    info_prefix = "kitti"
    info_path = f'{data_path}/{info_prefix}_infos_train.pkl'

    dataset_cfg = dict(
        type=dataset_class_name, data_root=data_path, ann_file=info_path)

    if dataset_class_name == 'KittiDataset':
        back_args = None
        dataset_cfg.update(
            test_mode=False,
            data_prefix=dict(pts=f"velodyne_reduced"),
            modality=dict(
                use_lidar=True,
            ),
            pipeline=[
                dict(
                    type='LoadPointsFromFile',
                    coord_type='LIDAR',
                    load_dim=4,
                    use_dim=4,
                    backend_args=back_args),
                dict(
                    type='LoadAnnotations3D',
                    with_bbox_3d=True,
                    with_label_3d=True,
                    backend_args=back_args)
            ],
            metainfo=dict(classes=['Car', 'Pedestrian', 'Cyclist'])
            )
    
    dataset = DATASETS.build(dataset_cfg)

KeyError: "class KittiDataset in mmdet3d/datasets/kitti_dataset.py: 'LoadPointsFromFile is not in the transform registry. Please check whether the value of LoadPointsFromFile is correct or it was registered as expected. More details can be found at https://mmengine.readthedocs.io/en/latest/advanced_tutorials/config.html#import-the-custom-module'"

[TiagoCortinhal]

I am facing the same issue, does anyone have a solution?

[sunjiahao1999]

import numpy as np 
from mmdet3d.registry import DATASETS
from mmengine.registry import init_default_scope  # Add this function
import mmcv
import mmengine
import mmdet3d



if __name__ == '__main__':
    init_default_scope('mmdet3d')  # Add
    dataset_class_name = 'KittiDataset'
    data_path = "./data/kitti"
    info_prefix = "kitti"
    info_path = f'{info_prefix}_infos_train.pkl'

    dataset_cfg = dict(
        type=dataset_class_name, data_root=data_path, ann_file=info_path)

    if dataset_class_name == 'KittiDataset':
        back_args = None
        dataset_cfg.update(
            test_mode=False,
            data_prefix=dict(pts=f"velodyne_reduced"),
            modality=dict(
                use_lidar=True,
            ),
            pipeline=[
                dict(
                    type='LoadPointsFromFile',
                    coord_type='LIDAR',
                    load_dim=4,
                    use_dim=4,
                    backend_args=back_args),
                dict(
                    type='LoadAnnotations3D',
                    with_bbox_3d=True,
                    with_label_3d=True,
                    backend_args=back_args)
            ],
            metainfo=dict(classes=['Car', 'Pedestrian', 'Cyclist'])
            )
    
    dataset = DATASETS.build(dataset_cfg)

[sunjiahao1999]

We need to init scope in mmdet3d, Otherwise it will use mmengine.