KeyError: 'CenterHead is not in the head registry'
darrenearl opened this issue · comments
(centerpoint) exinova@exinova-B560M-AORUS-PRO-AX:/media/exinova/ssd/CenterPoint-master_py$ python tools/train.py configs/nusc/pp/nusc_centerpoint_pp_02voxel_two_pfn_10sweep.py
2022-12-15 16:35:21,463 - INFO - Distributed training: False
2022-12-15 16:35:21,463 - INFO - torch.backends.cudnn.benchmark: False
2022-12-15 16:35:21,484 - INFO - Finish RPN Initialization
Traceback (most recent call last):
File "tools/train.py", line 168, in
main()
File "tools/train.py", line 141, in main
model = build_detector(cfg.model, train_cfg=cfg.train_cfg, test_cfg=cfg.test_cfg)
File "/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/models/builder.py", line 53, in build_detector
return build(cfg, DETECTORS, dict(train_cfg=train_cfg, test_cfg=test_cfg))
File "/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/models/builder.py", line 21, in build
return build_from_cfg(cfg, registry, default_args)
File "/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/utils/registry.py", line 82, in build_from_cfg
return obj_cls(**args)
File "/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/models/detectors/point_pillars.py", line 18, in init
reader, backbone, neck, bbox_head, train_cfg, test_cfg, pretrained
File "/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/models/detectors/single_stage.py", line 25, in init
self.bbox_head = builder.build_head(bbox_head)
File "/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/models/builder.py", line 45, in build_head
return build(cfg, HEADS)
File "/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/models/builder.py", line 21, in build
return build_from_cfg(cfg, registry, default_args)
File "/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/utils/registry.py", line 70, in build_from_cfg
"{} is not in the {} registry".format(obj_type, registry.name)
KeyError: 'CenterHead is not in the head registry'
what is wrong about this ? do anyone knowns?
it seems you are using the Det3D repo instead of the current one
"/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/models/detectors/point_pillars.py""
it seems you are using the Det3D repo instead of the current one
"/media/exinova/ssd/CenterPoint-master_py/Det3D/det3d/models/detectors/point_pillars.py""
but when i uninstall det3d, error also happend:
(centerpoint) exinova@exinova-B560M-AORUS-PRO-AX:/media/exinova/ssd/CenterPoint-master_py$ python tools/train.py configs/nusc/pp/nusc_centerpoint_pp_02voxel_two_pfn_10sweep.py
Traceback (most recent call last):
File "tools/train.py", line 14, in
from det3d.datasets import build_dataset
ModuleNotFoundError: No module named 'det3d'
add centerpoint's path to pythonpath