From ebd5ca4d478b1f5018193dca53be5916713a7504 Mon Sep 17 00:00:00 2001 From: "Z. W" <104838600+Nickory@users.noreply.github.com> Date: Tue, 17 Mar 2026 11:31:27 +0800 Subject: [PATCH] Refactor RMSE and RMSE log calculations --- test/metrics_st.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/metrics_st.py b/test/metrics_st.py index eda2acc..ddfda73 100644 --- a/test/metrics_st.py +++ b/test/metrics_st.py @@ -26,10 +26,10 @@ def compute_depth_metrics(gt, pred, mask=None, median_align=False): a3 = (thresh < 1.25 ** 3).float().mean() rmse = (gt - pred) ** 2 - rmse = torch.sqrt(rmse).mean() + rmse = torch.sqrt(rmse.mean()) rmse_log = (torch.log10(gt) - torch.log10(pred)) ** 2 - rmse_log = torch.sqrt(rmse_log).mean() + rmse_log = torch.sqrt(rmse_log.mean()) abs_ = torch.mean(torch.abs(gt - pred))