浏览代码

修复features_align函数参数传递错误导致cuda报错bug

lstrlq 5 月之前
父节点
当前提交
f62d0e4c3d
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      models/line_detect/trainer.py

+ 1 - 1
models/line_detect/trainer.py

@@ -215,7 +215,7 @@ class Trainer(BaseTrainer):
         if 'lines' in result:
             # 用自己写的函数画线段
             # line_image = draw_lines(boxed_image, result['lines'], color='red', width=3)
-            print(f"shape of linescore:{result['liness_scores'].shape}")
+            print(f"shape of linescore:{result['lines_scores'].shape}")
             scores = result['lines_scores'].mean(dim=1)  # shape: [31]
 
             line_image = draw_lines_with_scores((img * 255).to(torch.uint8),  result['lines'],scores, width=3, cmap='jet')