diff --git a/monocon/mmdet3d/models/dense_heads/monocon_head.py b/monocon/mmdet3d/models/dense_heads/monocon_head.py index 824f8c3..4f8c784 100755 --- a/monocon/mmdet3d/models/dense_heads/monocon_head.py +++ b/monocon/mmdet3d/models/dense_heads/monocon_head.py @@ -27,7 +27,7 @@ class MonoConHead(nn.Module): num_alpha_bins=12, max_objs=30, vector_regression_level=1, - pred_bbox2d=False, + pred_bbox2d=True, loss_center_heatmap=None, loss_wh=None, loss_offset=None, diff --git a/monocon/mmdet3d/models/dense_heads/monocon_head_inference.py b/monocon/mmdet3d/models/dense_heads/monocon_head_inference.py index bb7517e..b000246 100755 --- a/monocon/mmdet3d/models/dense_heads/monocon_head_inference.py +++ b/monocon/mmdet3d/models/dense_heads/monocon_head_inference.py @@ -23,6 +23,7 @@ class MonoConHeadInference(MonoConHead): def __init__(self, *args, **kwargs): super(MonoConHeadInference, self).__init__(*args, **kwargs) + self.pred_bbox2d = False self.wh_head = None self.offset_head = None self.kpt_heatmap_head = None