zhaoyinghan 1 kuukausi sitten
vanhempi
commit
9e6e71360d
1 muutettua tiedostoa jossa 1 lisäystä ja 0 poistoa
  1. 1 0
      models/line_detect/loi_heads.py

+ 1 - 0
models/line_detect/loi_heads.py

@@ -1406,6 +1406,7 @@ class RoIHeads(nn.Module):
                         loss_ins = compute_ins_loss(feature_logits, ins_proposals, gt_inses,ins_pos_matched_idxs)
                         total_loss, loss_arc_equation, loss_arc_ends = compute_arc_equation_loss(arc_equation,
                                                                                                  ins_proposals,
+
                                                                                                  gt_mask_ends,
                                                                                                  gt_mask_params,
                                                                                                  ins_pos_matched_idxs,