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):
mlvl_attr_scores)
bboxes, scores, labels, dir_scores, attrs = results
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)
# 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:
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