Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "04cd6adf8ca72f8fc99873ef03a5abce69ee6955"
Unverified Commit cec2d8b0 authored by Liang Liu's avatar Liang Liu Committed by GitHub
Browse files

[Fix]: code weights for anchor3d_head (#173)

parent c84285da
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet3d/models/dense_heads/anchor3d_head.py
View file @ cec2d8b0
...@@ -244,7 +244,7 @@ class Anchor3DHead(nn.Module, AnchorTrainMixin): ...@@ -244,7 +244,7 @@ class Anchor3DHead(nn.Module, AnchorTrainMixin):
if num_pos > 0: if num_pos > 0:
code_weight = self.train_cfg.get('code_weight', None) code_weight = self.train_cfg.get('code_weight', None)
if code_weight: if code_weight:
bbox_weights = bbox_weights * bbox_weights.new_tensor( pos_bbox_weights = pos_bbox_weights * bbox_weights.new_tensor(
code_weight) code_weight)
if self.diff_rad_by_sin: if self.diff_rad_by_sin:
pos_bbox_pred, pos_bbox_targets = self.add_sin_difference( pos_bbox_pred, pos_bbox_targets = self.add_sin_difference(
......
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