From d29bc650c3bab3d5159967c80a34222a7fd9b0b2 Mon Sep 17 00:00:00 2001 From: Jiao77 Date: Thu, 20 Nov 2025 05:18:47 +0800 Subject: [PATCH] change some problem 7 --- tools/diffusion/ic_layout_diffusion_optimized.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/diffusion/ic_layout_diffusion_optimized.py b/tools/diffusion/ic_layout_diffusion_optimized.py index ec5fa56..dd80cad 100644 --- a/tools/diffusion/ic_layout_diffusion_optimized.py +++ b/tools/diffusion/ic_layout_diffusion_optimized.py @@ -429,9 +429,9 @@ def manhattan_post_process(image, threshold=0.5): # 二值化 binary = (image > threshold).float() - # 形态学操作强化直角特征 - kernel_h = torch.tensor([[[[1,1,1]]]], device=device) - kernel_v = 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, dtype=image.dtype) # 水平和垂直增强 horizontal = F.conv2d(binary, kernel_h, padding=(0,1))