单通道提取4个点，有较好效果

models/line_detect/loi_heads.py

@@ -14,7 +14,7 @@ from collections import OrderedDict
 
 from models.line_detect.heads.head_losses import point_inference, compute_point_loss, line_iou_loss, \
     lines_point_pair_loss, features_align, line_inference, compute_arc_loss, arc_inference, compute_circle_loss, \
-    circle_inference, compute_circle_extra_losses
+    circle_inference
 
 
 def fastrcnn_loss(class_logits, box_regression, labels, regression_targets):
@@ -1370,7 +1370,8 @@ class RoIHeads(nn.Module):
                     print(f'start to compute circle_loss')
 
                     loss_circle = compute_circle_loss(feature_logits, circle_proposals, gt_circles, circle_pos_matched_idxs)
-                    loss_circle_extra=compute_circle_extra_losses(feature_logits, circle_proposals, gt_circles, circle_pos_matched_idxs)
+
+                    # loss_circle_extra=compute_circle_extra_losses(feature_logits, circle_proposals, gt_circles, circle_pos_matched_idxs)
 
                 if loss_circle is None:
                     print(f'loss_circle is None111')
@@ -1400,7 +1401,7 @@ class RoIHeads(nn.Module):
                         loss_circle = compute_circle_loss(feature_logits, circle_proposals, gt_circles,
                                                         circle_pos_matched_idxs)
 
-                        loss_circle_extra = compute_circle_extra_losses(feature_logits, circle_proposals, gt_circles,circle_pos_matched_idxs)
+                        # loss_circle_extra = compute_circle_extra_losses(feature_logits, circle_proposals, gt_circles,circle_pos_matched_idxs)
 
                     if loss_circle is None:
                         print(f'loss_circle is None111')