|
|
@@ -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')
|