修复batchsize大于1时 报错bug

models/line_detect/loi_heads.py

@@ -500,10 +500,10 @@ def heatmaps_to_lines(maps, rois):
 def lines_features_align(features, proposals, img_size):
     print(f'lines_features_align features:{features.shape}')
 
-
+    align_feat_list = []
     for feat, proposals_per_img  in zip(features,proposals):
         # print(f'lines_features_align feat:{feat.shape}, proposals_per_img:{proposals_per_img.shape}')
-        align_feat_list=[]
+
         feat=feat.unsqueeze(0)
         for proposal in proposals_per_img:
             align_feat = torch.zeros_like(feat)

models/line_detect/train.yaml

@@ -1,6 +1,6 @@
 io:
   logdir: train_results
-  datadir: \\192.168.50.222/share/zyh/202507/a_dataset
+  datadir: /data/share/zyh/202507/a_dataset
   data_type: rgb
 #  datadir: D:\python\PycharmProjects\data_20250223\0423_
 #  datadir: I:\datasets\wirenet_1000
@@ -11,7 +11,7 @@ io:
 train_params:
   resume_from:
   num_workers: 8
-  batch_size: 1
+  batch_size: 4
   max_epoch: 80000
   augmentation: True
   optim: