INNER CODE UNIT · Python
pred_cam_points_b3hw
nianticlabs/simplerecon · losses.py:162
pred_cam_points_b3hw = src_cam_points_b3N.view(-1, 3, depth_height,
depth_width)
pred_src_depths_b1hw = pred_cam_points_b3hw[:, 2:]
depth_diff_b1hw = torch.abs(
torch.log(src_depth_sampled_b1hw) -
torch.log(pred_src_depths_b1hw)
).masked_select(valid_mask_b1hw)
depth_loss = depth_diff_b1hw.nanmean()
return depth_loss
def forward(
self,
depth_pred_b1hw,
cur_depth_b1hw,
src_depth_bk1hw,