open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io
Apache License 2.0
28.52k stars 9.29k forks source link

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

Open deadeye8858 opened 1 month ago

deadeye8858 commented 1 month ago

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'

suliangxu commented 1 month ago

您好,这个问题解决了吗