nickgkan / butd_detr

Code for the ECCV22 paper "Bottom Up Top Down Detection Transformers for Language Grounding in Images and Point Clouds"
Other
74 stars 11 forks source link

the confidence threshold you set for the pretrained detector #30

Closed linhaojia13 closed 1 year ago

linhaojia13 commented 1 year ago

As you mentioned in the papers:

... Group-Free detector [33] for 3D point clouds pre-trained on a vocabulary of 485 object categories on ScanNet [8]. The detected box proposals that surpass a confidence threshold are encoded using a box proposal encoder ...

Could you please tell me the confidence threshold you set for the pretrained Group-Free detector?

ayushjain1144 commented 1 year ago

This is what we used:

eval_config_dict = {'remove_empty_box': True, 'use_3d_nms': True, 'nms_iou': 0.25,
'use_old_type_nms': False, 'cls_nms': False, 'per_class_proposal': False,
'conf_thresh': 0.5, 'dataset_config': DC}

Note that we didn't really play around with these, so they may not be optimal.

linhaojia13 commented 1 year ago

This is what we used:

eval_config_dict = {'remove_empty_box': True, 'use_3d_nms': True, 'nms_iou': 0.25,
'use_old_type_nms': False, 'cls_nms': False, 'per_class_proposal': False,
'conf_thresh': 0.5, 'dataset_config': DC}

Note that we didn't really play around with these, so they may not be optimal.

Thank you very much!