Inference
/
MonoCon
mirror of https://github.com/Xianpeng919/MonoCon.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monocon/mmdet3d/models/dense_heads/monocon_head.py
View File

@ -27,7 +27,7 @@ class MonoConHead(nn.Module):
num_alpha_bins=12, num_alpha_bins=12,
max_objs=30, max_objs=30,
vector_regression_level=1, vector_regression_level=1,
pred_bbox2d=False, pred_bbox2d=True,
loss_center_heatmap=None, loss_center_heatmap=None,
loss_wh=None, loss_wh=None,
loss_offset=None, loss_offset=None,

1
monocon/mmdet3d/models/dense_heads/monocon_head_inference.py
View File

@ -23,6 +23,7 @@ class MonoConHeadInference(MonoConHead):
def __init__(self, def __init__(self,
*args, **kwargs): *args, **kwargs):
super(MonoConHeadInference, self).__init__(*args, **kwargs) super(MonoConHeadInference, self).__init__(*args, **kwargs)
self.pred_bbox2d = False
self.wh_head = None self.wh_head = None
self.offset_head = None self.offset_head = None
self.kpt_heatmap_head = None self.kpt_heatmap_head = None