"src/rpc/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "27cad329db9920b5f6f7d420a50c9e3f78d12485"
Unverified Commit e108340c authored by twang's avatar twang Committed by GitHub
Browse files

Fix the origin setting caused by an adjustment from #283 (#484)

parent f7356f4b
...@@ -678,10 +678,12 @@ class FCOSMono3DHead(AnchorFreeMono3DHead): ...@@ -678,10 +678,12 @@ class FCOSMono3DHead(AnchorFreeMono3DHead):
mlvl_attr_scores) mlvl_attr_scores)
bboxes, scores, labels, dir_scores, attrs = results bboxes, scores, labels, dir_scores, attrs = results
attrs = attrs.to(labels.dtype) # change data type to int attrs = attrs.to(labels.dtype) # change data type to int
bboxes = input_meta['box_type_3d'](bboxes, box_dim=self.bbox_code_size) bboxes = input_meta['box_type_3d'](
bboxes, box_dim=self.bbox_code_size, origin=(0.5, 0.5, 0.5))
# Note that the predictions use origin (0.5, 0.5, 0.5) # Note that the predictions use origin (0.5, 0.5, 0.5)
# Due to the ground truth centers2d are the gravity center of objects # Due to the ground truth centers2d are the gravity center of objects
# The center has been transformed when computing bev bbox!!!! # v0.10.0 fix inplace operation to the input tensor of cam_box3d
# So here we also need to add origin=(0.5, 0.5, 0.5)
if not self.pred_attrs: if not self.pred_attrs:
attrs = None attrs = None
......
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