添加keypoinits生成线图功能

models/dataset_tool.py

@@ -236,17 +236,17 @@ def line_boxes(target):
             keypoints.append([a[1], b[1]])
 
             if a[1] > b[1]:
-                ymax = a[1] + 1
-                ymin = b[1] - 1
+                ymax = a[1] + 10
+                ymin = b[1] - 10
             else:
-                ymin = a[1] - 1
-                ymax = b[1] + 1
+                ymin = a[1] - 10
+                ymax = b[1] + 10
             if a[0] > b[0]:
-                xmax = a[0] + 1
-                xmin = b[0] - 1
+                xmax = a[0] + 10
+                xmin = b[0] - 10
             else:
-                xmin = a[0] - 1
-                xmax = b[0] + 1
+                xmin = a[0] - 10
+                xmax = b[0] + 10
             boxs.append([ymin, xmin, ymax, xmax])
 
     return torch.tensor(boxs), torch.tensor(keypoints)