Function compute_metrics_for_each_image

tools.py:160–173

(metric_func)

# a wrapper to compute metrics for each image individually
def compute_metrics_for_each_image(metric_func):
    def wrapper(depth_est, depth_gt, mask, *args):
        batch_size = depth_gt.shape[0]
        results = []
        # compute result one by one
        for idx in range(batch_size):
            ret = metric_func(depth_est[idx], depth_gt[idx], mask[idx], *args)
            if torch.isnan(ret):
                results.append(torch.zeros_like(ret))
            else:
                results.append(ret)
        return torch.stack(results).mean()

    return wrapper


@torch.no_grad()