Baidu-netdisk-AI-Image-processing-Challenge-demoire icon indicating copy to clipboard operation
Baidu-netdisk-AI-Image-processing-Challenge-demoire copied to clipboard

Published 20 hours ago • verified publisher icon zdyshine

大神 ，计算损失函数时，下面分别获取rgb通道是否写的有问题

Open Testhjf opened this issue 1 year ago • 0 comments

loss_l1_r = loss_criterion(out_tensor[0, :, :], tensor_ref[0, :, :])
loss_l1_g = loss_criterion(out_tensor[1, :, :], tensor_ref[1, :, :]) loss_l1_b = loss_criterion(out_tensor[2, :, :], tensor_ref[2, :, :])

应该是写成loss_l1_r = loss_criterion(out_tensor[:,0, :, :], tensor_re[:,0, :, :],) 吧

Testhjf avatar Apr 29 '23 06:04 Testhjf