Merge pull request traveller59#161 from dashidhy/patch-1

fixed a small bug in box coder builder
Hqss · Apr 17, 2019 · c7bb659 · c7bb659
2 parents 8eaa9bf + 5aa59f8
commit c7bb659
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/second/pytorch/builder/box_coder_builder.py b/second/pytorch/builder/box_coder_builder.py
@@ -19,7 +19,7 @@ def build(box_coder_config):
     """
     box_coder_type = box_coder_config.WhichOneof('box_coder')
     if box_coder_type == 'ground_box3d_coder':
-        cfg = box_coder_config.bev_box_coder
+        cfg = box_coder_config.ground_box3d_coder
         return GroundBox3dCoderTorch(cfg.linear_dim, cfg.encode_angle_vector)
     elif box_coder_type == 'bev_box_coder':
         cfg = box_coder_config.bev_box_coder