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 03:11:33 +08:00
parent 26763fa75c
commit bacf8cd69d
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

@@ -108,9 +108,9 @@ class EdgeAwareLoss(nn.Module):
def __init__(self): def __init__(self):
super().__init__() super().__init__()
# 注册为缓冲区以避免重复创建 # 注册为缓冲区以避免重复创建，并指定为浮点类型
self.register_buffer('sobel_x', torch.tensor([[[[-1,0,1],[-2,0,2],[-1,0,1]]]])) self.register_buffer('sobel_x', torch.tensor([[[[-1.,0.,1.],[-2.,0.,2.],[-1.,0.,1.]]]]))
self.register_buffer('sobel_y', torch.tensor([[[[-1,-2,-1],[0,0,0],[1,2,1]]]])) self.register_buffer('sobel_y', torch.tensor([[[[-1.,-2.,-1.],[0.,0.,0.],[1.,2.,1.]]]]))
def forward(self, pred, target): def forward(self, pred, target):
# 原始MSE损失 # 原始MSE损失