only arc boxes is ok

models/line_detect/line_dataset.py

@@ -655,6 +655,6 @@ def get_boxes_lines(objs, shape):
 
 if __name__ == '__main__':
     path = r'/data/share/zyh/master_dataset/dataset_net/pokou_251115_251121/a_dataset'
-    dataset = LineDataset(dataset_path=path, dataset_type='train', augmentation=False, data_type='jpg')
-    for i in range(100):
-         dataset.show(i, show_type='Original_mask')
+    dataset = LineDataset(dataset_path=path, dataset_type='val', augmentation=False, data_type='jpg')
+    for i in range(len(dataset)):
+         dataset.show(i, show_type='boxes')

models/line_detect/trainer.py

@@ -621,6 +621,7 @@ class Trainer(BaseTrainer):
             if phase == 'train':
                 optimizer.zero_grad()
                 loss.backward()
+                torch.nn.utils.clip_grad_norm_(model.parameters(), 5.0)
                 optimizer.step()
             self.writer_loss(loss_dict, global_step, phase=phase)
             global_step += 1