Skip to content

GitLab

Unverified Commit aad3339b authored by Xiangxu-0103's avatar Xiangxu-0103 Committed by GitHub
Browse files

fix roi_extractors.build bug (#1460)

parent 4422eaab
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet3d/models/builder.py
View file @ aad3339b
...@@ -46,7 +46,7 @@ def build_neck(cfg): ...@@ -46,7 +46,7 @@ def build_neck(cfg):
def build_roi_extractor(cfg): def build_roi_extractor(cfg):
"""Build RoI feature extractor.""" """Build RoI feature extractor."""
if cfg['type'] in NECKS._module_dict.keys(): if cfg['type'] in ROI_EXTRACTORS._module_dict.keys():
return ROI_EXTRACTORS.build(cfg) return ROI_EXTRACTORS.build(cfg)
else: else:
return MMDET_ROI_EXTRACTORS.build(cfg) return MMDET_ROI_EXTRACTORS.build(cfg)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment