open-mmlab/mmdetection

"CoDINOHead is not registered" occurs when running the codetr model.

deadeye8858 opened this issue · 1 comments

The error occurred during training of the CODETR model.
File "j:/迅雷下载/mmdetection-3.3.0/mmdetection-3.3.0/tools/train.py", line 123, in
main()
File "j:/迅雷下载/mmdetection-3.3.0/mmdetection-3.3.0/tools/train.py", line 112, in main
runner = Runner.from_cfg(cfg)
File "C:\Users\322.conda\envs\fz4\lib\site-packages\mmengine\runner\runner.py", line 445, in from_cfg
runner = cls(
File "C:\Users\322.conda\envs\fz4\lib\site-packages\mmengine\runner\runner.py", line 412, in init
self.model = self.build_model(model)
File "C:\Users\322.conda\envs\fz4\lib\site-packages\mmengine\runner\runner.py", line 819, in build_model
model = MODELS.build(model)
File "C:\Users\322.conda\envs\fz4\lib\site-packages\mmengine\registry\registry.py", line 570, in build
return self.build_func(cfg, *args, **kwargs, registry=self)
File "C:\Users\322.conda\envs\fz4\lib\site-packages\mmengine\registry\build_functions.py", line 232, in build_model_from_cfg
return build_from_cfg(cfg, registry, default_args)
File "C:\Users\322.conda\envs\fz4\lib\site-packages\mmengine\registry\build_functions.py", line 121, in build_from_cfg
obj = obj_cls(**args) # type: ignore
File "J:\迅雷下载\mmdetection-3.3.0\mmdetection-3.3.0\projects\CO-DETR\codetr\codetr.py", line 55, in init
self.query_head = MODELS.build(query_head)
File "C:\Users\322.conda\envs\fz4\lib\site-packages\mmengine\registry\registry.py", line 570, in build
return self.build_func(cfg, *args, **kwargs, registry=self)
File "C:\Users\322.conda\envs\fz4\lib\site-packages\mmengine\registry\build_functions.py", line 232, in build_model_from_cfg
return build_from_cfg(cfg, registry, default_args)
File "C:\Users\322.conda\envs\fz4\lib\site-packages\mmengine\registry\build_functions.py", line 100, in build_from_cfg
raise KeyError(
KeyError: 'CoDINOHead is not in the mmdet::model registry. Please check whether the value of CoDINOHead 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'

您好,这个问题解决了吗