dataset修改，box大小边界

models/line_detect/dataset_LD.py

@@ -54,12 +54,12 @@ def line_boxes1(target):
             else:
                 xmin = a[0] - 10
                 xmax = b[0] + 10
-            boxs.append([ymin, xmin, ymax, xmax])
+            boxs.append([min(ymin,0), min(xmin,0), max(ymax,512), max(xmax, 0)])
 
     # print(f'box:{boxs}')
-    if boxs == []:
-        print(f'box:{boxs}')
-        print(f'target:{target}')
+    # if boxs == []:
+    #     print(f'box:{boxs}')
+    #     print(f'target:{target}')
 
     return torch.tensor(boxs)