gt_keypoint_depth_weights = tf.constant(
          np.array([[1.0, 1.0, 1.0, 1.0, 1.0],
                    [float('nan'), 0.0, 1.0, 0.0, 0.0],
                    [1.0, 1.0, 1.0, 1.0, 1.0],
                    [1.0, 1.0, 0.5, 1.0, 1.0]]),