调试resnet50识别single point ,效果良好

models/line_detect/heads/head_losses.py

@@ -232,8 +232,8 @@ def generate_gaussian_heatmaps(xs, ys, heatmap_size,num_points=2, sigma=2.0, dev
 
     assert xs.shape == ys.shape, "x and y must have the same shape"
     print(f'xs:{xs.shape}')
-    xs=xs.squeeze(1)
-    ys = ys.squeeze(1)
+    # xs=xs.squeeze(1)
+    # ys = ys.squeeze(1)
     print(f'xs1:{xs.shape}')
     N = xs.shape[0]
     print(f'N:{N},num_points:{num_points}')

models/line_detect/line_detect.py

@@ -187,7 +187,7 @@ class LineDetect(BaseDetectionNet):
 
         if point_predictor is None and detect_point:
         #     keypoint_dim_reduced = 512  # == keypoint_layers[-1]
-            point_predictor = PointPredictor(in_channels=128)
+            point_predictor = PointPredictor(in_channels=256)
 
         if detect_arc and arc_head is None:
             layers = tuple(num_points for _ in range(8))
@@ -389,7 +389,11 @@ def linedetect_newresnet50fpn(
 
     anchor_generator =  AnchorGenerator(sizes=anchor_sizes, aspect_ratios=aspect_ratios)
 
-    model = LineDetect(backbone, num_classes,min_size=size,max_size=size, num_points=num_points, rpn_anchor_generator=anchor_generator, box_roi_pool=roi_pooler, **kwargs)
+    model = LineDetect(backbone, num_classes,min_size=size,max_size=size, num_points=num_points, rpn_anchor_generator=anchor_generator, box_roi_pool=roi_pooler,
+                       detect_point=True,
+                       detect_line=False,
+                       detect_arc=False,
+                       **kwargs)
 
 
 

models/line_detect/train.yaml

@@ -1,13 +1,13 @@
 io:
   logdir: train_results
-<<<<<<< HEAD
-  datadir: /data/share/rlq/datasets/250718caisegangban
+
+#  datadir: /data/share/rlq/datasets/250718caisegangban
 #  datadir: /data/share/zjh/Dataset_correct_xanylabel_tiff
-=======
+
 #  datadir: /data/share/rlq/datasets/250718caisegangban
->>>>>>> 55050f283a5d85b7c3731e296d70483d8f7258e2
-#  datadir: /data/share/rlq/datasets/singepoint_Dataset0709_2
-  datadir: \\192.168.50.222/share/zyh/arc/a_dataset
+
+  datadir: /data/share/rlq/datasets/singepoint_Dataset0709_2
+#  datadir: \\192.168.50.222/share/zyh/arc/a_dataset
 #  datadir: \\192.168.50.222/share/rlq/datasets/singepoint_Dataset0709_2
 #  datadir: \\192.168.50.222/share/rlq/datasets/250718caisegangban
   data_type: rgb