fix a bug for validation during training

pull/30/head
xianpeng 2022-02-26 12:50:51 -05:00
parent a31b4690f6
commit 1f32c66c5f
2 changed files with 2 additions and 1 deletions

View File

@ -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,

View File

@ -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