Function rgb_loss

script/feature/direct_feature_matching.py:95–100

Compute RGB MSE Loss, original from NeRF Paper

(rgb, target, extras)

    return features, pose

def rgb_loss(rgb, target, extras):
    ''' Compute RGB MSE Loss, original from NeRF Paper '''
    # Compute MSE loss between predicted and true RGB.
    img_loss = img2mse(rgb, target)
    loss = img_loss
    return loss

def normalize_features(tensor, value_range=None, scale_each: bool = False):
    ''' Find unit norm of channel wise feature