修改line的box框大小

models/line_detect/line_dataset.py

@@ -183,10 +183,10 @@ def get_boxes_lines(target):
             line_point_pairs.append([a[1], a[0]])
             line_point_pairs.append([b[1], b[0]])
 
-            xmin = min(a[0], b[0]) - 1
-            xmax = max(a[0], b[0]) + 1
-            ymin = min(a[1], b[1]) - 1
-            ymax = max(a[1], b[1]) + 1
+            xmin =max(0 ,(min(a[0], b[0]) - 6))
+            xmax =min(511, (max(a[0], b[0]) + 6))
+            ymin =max(0, (min(a[1], b[1]) - 6))
+            ymax =min(511, (max(a[1], b[1]) + 6))
 
             boxs.append([ymin, xmin, ymax, xmax])