Skip to content

GitLab

Unverified Commit 1145f836 authored by Shaoshuai Shi's avatar Shaoshuai Shi Committed by GitHub
Browse files

bugfixed: typo for using multihead config, not critical (#352)

parent a6bb3580
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pcdet/models/dense_heads/target_assigner/axis_aligned_target_assigner.py
View file @ 1145f836
...@@ -25,7 +25,7 @@ class AxisAlignedTargetAssigner(object): ...@@ -25,7 +25,7 @@ class AxisAlignedTargetAssigner(object):
self.unmatched_thresholds[config['class_name']] = config['unmatched_threshold'] self.unmatched_thresholds[config['class_name']] = config['unmatched_threshold']
self.use_multihead = model_cfg.get('USE_MULTIHEAD', False) self.use_multihead = model_cfg.get('USE_MULTIHEAD', False)
self.seperate_multihead = model_cfg.get('SEPERATE_MULTIHEAD', False) self.seperate_multihead = model_cfg.get('SEPARATE_MULTIHEAD', False)
if self.seperate_multihead: if self.seperate_multihead:
rpn_head_cfgs = model_cfg.RPN_HEAD_CFGS rpn_head_cfgs = model_cfg.RPN_HEAD_CFGS
self.gt_remapping = {} self.gt_remapping = {}
......
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