search

luogen1996 / OneTeacher

GNU General Public License v3.0
78 stars 9 forks source link

About non_max_suppression_pseudo_decouple_multi_view #8

Open huyennt-1104 opened 1 year ago

huyennt-1104 commented 1 year ago 
if isinstance(out,list):

pseudo_boxes_reg, pseudo_boxes_cls = non_max_suppression_pseudo_decouple_multi_view(out, hyp[ 'bbox_threshold'], hyp['cls_threshold'], multi_label=True ,agnostic=single_cls, )

else:

pseudo_boxes_reg,pseudo_boxes_cls = non_max_suppression_pseudo_decouple(out, hyp['bbox_threshold'],hyp['cls_threshold'], multi_label=True, agnostic=single_cls )

https://github.com/luogen1996/OneTeacher/blob/9a0e798fbe646cbab19e7df00cd27ebc817c36a0/train_semi.py#L527

When I run code, it always run at non_max_suppression_pseudo_decouple. How type of model, how type config can output a list and run non_max_suppression_pseudo_decouple_multi_view?