INNER CODE UNIT · Python
get_error_for_pair
nianticlabs/simplerecon · losses.py:136
def get_error_for_pair(self,
depth_pred_b1hw,
cur_depth_b1hw,
src_depth_b1hw,
cur_invK_b44,
src_K_b44,
cur_world_T_cam_b44,
src_cam_T_world_b44):
depth_height, depth_width = cur_depth_b1hw.shape[2:]
valid_mask_b1hw, src_depth_sampled_b1hw = self.get_valid_mask(
cur_depth_b1hw,
src_depth_b1hw,
cur_invK_b44,
src_K_b44,
cur_world_T_cam_b44,
src_cam_T_world_b44,