admin 管理员组

文章数量: 1086019

【code

def edge_aware_loss_v2(rgb, disp,skymask=None):"""Computes the smoothness loss for a disparity imageThe color image is used for edge-aware smoothness"""mean_disp = disp.mean(1, True).mean(2, True)#行&列的均值disp = disp / (mean_disp + 1e-7)#归一化处理grad_disp_x = torch.abs(disp[:, :, :-1, :] - disp[:, :, 1:, :])#x轴梯度grad_disp_y = torch.abs(disp[:, :-1, :, :] - disp[:, 1:, :, :]))#y轴梯度grad_rgb_x = torch.mean(torch.abs(rgb[:, :, :-1, :] - rgb[:, :, 1:, :]), 3, keepdim=True)grad_rgb_y = torch.mean(torch.abs(rgb[:, :-1, :, :] - rgb[:, 1:, :, :]), 3, keepdim=True)grad_disp_x *= torch.exp(-grad_rgb_x)grad_disp_y *= torch.exp(-grad_rgb_y)# mask=torch.ones_like(disp)if skymask is not None:grad_disp_x+=skymask[:,:,:-1,:]*grad_disp_xgrad_disp_y+=skymask[:,:-1,:,:]*grad_disp_yreturn grad_disp_x.mean() + grad_disp_y.mean()

参考链接:边缘感知平滑损失

本文标签： Code

版权声明：本文标题：【code 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/p/1700349938a408498.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。