调试华沿数据集,box有效果

models/line_detect/line_dataset.py

@@ -147,7 +147,7 @@ class LineDataset(BaseDataset):
         if show_type=='all':
             boxed_image = draw_bounding_boxes((img * 255).to(torch.uint8), target["boxes"],
                                                   colors="yellow", width=1)
-            circle=target['circle']
+            circle=target['circles']
             print(f'taget circle:{circle.shape}')
             keypoint_img=draw_keypoints(boxed_image,circle,colors='red',width=3)
             plt.imshow(keypoint_img.permute(1, 2, 0).numpy())
@@ -257,13 +257,13 @@ def get_boxes_lines(objs,shape):
             points=sort_points_clockwise(obj['points'])
             circle_4points.append(points)
 
-            xmin = max(obj['xmin'] - 6, 0)
+            xmin = max(obj['xmin'] - 40, 0)
 
-            xmax = min(obj['xmax'] + 6, w)
+            xmax = min(obj['xmax'] + 40, w)
 
-            ymin = max(obj['ymin'] - 6, 0)
+            ymin = max(obj['ymin'] - 40, 0)
 
-            ymax = min(obj['ymax'] + 6, h)
+            ymax = min(obj['ymax'] + 40, h)
 
             boxes.append([xmin, ymin, xmax, ymax])
 
@@ -303,6 +303,6 @@ def get_boxes_lines(objs,shape):
     return boxes,line_point_pairs,points,line_mask,circle_4points, labels
 
 if __name__ == '__main__':
-    path=r"\\192.168.50.222/share/zyh/data/rgb_4point/a_dataset"
+    path=r'/data/share/zyh/master_dataset/circle/huayan_circle/a_dataset'
     dataset= LineDataset(dataset_path=path, dataset_type='train',augmentation=False, data_type='jpg')
-    dataset.show(99,show_type='all')
+    dataset.show(9,show_type='all')

models/line_detect/train.yaml

@@ -7,7 +7,7 @@ io:
 #  datadir: /data/share/rlq/datasets/250718caisegangban
 
 #  datadir: /data/share/rlq/datasets/singepoint_Dataset0709_2
-  datadir: /home/zhaoyinghan/py_ws/data/circle/a_dataset
+  datadir: /data/share/zyh/master_dataset/circle/huayan_circle/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