fix a bug for validation during training
parent
a31b4690f6
commit
1f32c66c5f
|
@ -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,
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue