Code

[YIXIN-M]

"""
def l1_loss_weight(network_output, gt):
image = gt.detach().cpu().numpy().transpose((1, 2, 0))
rgb_raw_gray = np.dot(image[..., :3], [0.2989, 0.5870, 0.1140])
sobelx = cv2.Sobel(rgb_raw_gray, cv2.CV_64F, 1, 0, ksize=5)
sobely = cv2.Sobel(rgb_raw_gray, cv2.CV_64F, 0, 1, ksize=5)
sobel_merge = np.sqrt(sobelx * sobelx + sobely * sobely) + 1e-10
sobel_merge = np.exp(sobel_merge)
sobel_merge /= np.max(sobel_merge)
sobel_merge = torch.from_numpy(sobel_merge)[None, ...].to(gt.device)

return torch.abs((network_output - gt) * sobel_merge).mean()

def l2_loss(network_output, gt):
return ((network_output - gt) ** 2).mean()
"""
Have these two functions（ l1_loss_weight and l2_loss）not been called in the code？