Jiao77/RoRD-Layout-Recognation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change some problem 7

Browse Source
This commit is contained in:
Jiao77
2025-11-20 05:18:47 +08:00
parent afd48c2d86
commit d29bc650c3
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tools/diffusion/ic_layout_diffusion_optimized.py
View File

@@ -429,9 +429,9 @@ def manhattan_post_process(image, threshold=0.5):
# 二值化 # 二值化
binary = (image > threshold).float() binary = (image > threshold).float()
# 形态学操作强化直角特征 # 形态学操作强化直角特征 - 使用浮点类型
kernel_h = torch.tensor([[[[1,1,1]]]], device=device) kernel_h = torch.tensor([[[[1.,1.,1.]]]], device=device, dtype=image.dtype)
kernel_v = torch.tensor([[[[1],[1],[1]]]], device=device) kernel_v = torch.tensor([[[[1.],[1.],[1.]]]], device=device, dtype=image.dtype)
# 水平和垂直增强 # 水平和垂直增强
horizontal = F.conv2d(binary, kernel_h, padding=(0,1)) horizontal = F.conv2d(binary, kernel_h, padding=(0,1))