重构ins_detect

models/line_detect/line_detect.py

@@ -155,7 +155,7 @@ class LineDetect(BaseDetectionNet):
             detect_point=detect_point,
             detect_line=detect_line,
             detect_arc=detect_arc,
-            detect_circle=detect_ins,
+            detect_ins=detect_ins,
         )
 
         if image_mean is None:

models/line_detect/loi_heads.py

@@ -557,7 +557,7 @@ class RoIHeads(nn.Module):
             detect_point=True,
             detect_line=False,
             detect_arc=False,
-            detect_circle=False,
+            detect_ins=False,
     ):
         super().__init__()
 
@@ -608,7 +608,7 @@ class RoIHeads(nn.Module):
         self.detect_point =detect_point
         self.detect_line =detect_line
         self.detect_arc =detect_arc
-        self.detect_circle=detect_circle
+        self.detect_ins=detect_ins
 
         self.channel_compress = nn.Sequential(
             nn.Conv2d(256, 8, kernel_size=1),
@@ -1321,7 +1321,7 @@ class RoIHeads(nn.Module):
                 losses.update(loss_arc)
                 print(f'losses:{losses}')
 
-        if self.has_ins and self.detect_circle:
+        if self.has_ins and self.detect_ins:
             print(f'roi_heads forward has_circle()!!!!')
             # print(f'labels:{labels}')
             circle_proposals = [p["boxes"] for p in result]