From 1f32c66c5faf7a5a4582dfbee427eadea2c00331 Mon Sep 17 00:00:00 2001 From: xianpeng Date: Sat, 26 Feb 2022 12:50:51 -0500 Subject: [PATCH] fix a bug for validation during training --- monocon/mmdet3d/models/dense_heads/monocon_head.py | 2 +- monocon/mmdet3d/models/dense_heads/monocon_head_inference.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) 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