def get_prediction(self, nms_pre=2000):
    refine_anchors = self.refine_anchor_list
    fam_cls_branch_list = self.s2anet_head_out[0]
    fam_reg_branch_list = self.s2anet_head_out[1]
    odm_cls_branch_list = self.s2anet_head_out[2]
    odm_reg_branch_list = self.s2anet_head_out[3]
    (pred_scores, pred_bboxes) = self.get_bboxes(odm_cls_branch_list, odm_reg_branch_list, refine_anchors, nms_pre, self.cls_out_channels, self.use_sigmoid_cls)
    return (pred_scores, pred_bboxes)
def get_prediction(self, nms_pre=2000):
    refine_anchors = self.refine_anchor_list
    fam_cls_branch_list = self.s2anet_head_out[0]
    fam_reg_branch_list = self.s2anet_head_out[1]
    odm_cls_branch_list = self.s2anet_head_out[2]
    odm_reg_branch_list = self.s2anet_head_out[3]
    (pred_scores, pred_bboxes) = self.get_bboxes(odm_cls_branch_list, odm_reg_branch_list, refine_anchors, nms_pre, self.cls_out_channels, self.use_sigmoid_cls)
    return (pred_scores, pred_bboxes)