search

HRNet / HRNet-Object-Detection

Object detection with multi-level representations generated from deep high-resolution representation learning (HRNetV2h). This is an official implementation for our TPAMI paper "Deep High-Resolution Representation Learning for Visual Recognition". https://arxiv.org/abs/1908.07919
Apache License 2.0
643 stars 98 forks source link

KeyError: 'HRNet is not in the backbone registry' #24

Closed YijianLiu closed 4 years ago

YijianLiu commented 4 years ago

I meet an error and I don't know how to solve it, could you help me? 2019-09-25 08:15:32,758 - INFO - Distributed training: True Traceback (most recent call last): File "tools/train.py", line 95, in main() File "tools/train.py", line 82, in main cfg.model, train_cfg=cfg.train_cfg, test_cfg=cfg.test_cfg) File "/home/cartur/.local/lib/python3.7/site-packages/mmdet-0.6rc0+unknown-py3.7.egg/mmdet/models/builder.py", line 51, in build_detector return build(cfg, DETECTORS, dict(train_cfg=train_cfg, test_cfg=test_cfg)) File "/home/cartur/.local/lib/python3.7/site-packages/mmdet-0.6rc0+unknown-py3.7.egg/mmdet/models/builder.py", line 31, in build return _build_module(cfg, registry, default_args) File "/home/cartur/.local/lib/python3.7/site-packages/mmdet-0.6rc0+unknown-py3.7.egg/mmdet/models/builder.py", line 23, in _build_module return obj_type(**args) File "/home/cartur/.local/lib/python3.7/site-packages/mmdet-0.6rc0+unknown-py3.7.egg/mmdet/models/detectors/faster_rcnn.py", line 25, in init pretrained=pretrained) File "/home/cartur/.local/lib/python3.7/site-packages/mmdet-0.6rc0+unknown-py3.7.egg/mmdet/models/detectors/two_stage.py", line 27, in init self.backbone = builder.build_backbone(backbone) File "/home/cartur/.local/lib/python3.7/site-packages/mmdet-0.6rc0+unknown-py3.7.egg/mmdet/models/builder.py", line 35, in build_backbone return build(cfg, BACKBONES) File "/home/cartur/.local/lib/python3.7/site-packages/mmdet-0.6rc0+unknown-py3.7.egg/mmdet/models/builder.py", line 31, in build return _build_module(cfg, registry, default_args) File "/home/cartur/.local/lib/python3.7/site-packages/mmdet-0.6rc0+unknown-py3.7.egg/mmdet/models/builder.py", line 15, in _build_module obj_type, registry.name)) KeyError: 'HRNet is not in the backbone registry'

wondervictor commented 4 years ago

In this repo, backbone registry is defined as "HighResolutionNet". https://github.com/HRNet/HRNet-Object-Detection/blob/e9261d58a54c2a7e69b81295d58d52c10ea5c791/configs/hrnet/faster_rcnn_hrnetv2p_w18_2x.py#L6

While in the official mmdetection. 'HRNet' is the backbone registry