From 21e3f92461e282f3909619d420582756605a7503 Mon Sep 17 00:00:00 2001 From: HTensor Date: Thu, 27 Apr 2023 19:15:49 +0800 Subject: [PATCH] changed metrics indicator --- IGEV-Stereo/train_stereo.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/IGEV-Stereo/train_stereo.py b/IGEV-Stereo/train_stereo.py index 00e2deb..feaf353 100644 --- a/IGEV-Stereo/train_stereo.py +++ b/IGEV-Stereo/train_stereo.py @@ -68,9 +68,10 @@ def sequence_loss(disp_preds, disp_init_pred, disp_gt, valid, loss_gamma=0.9, ma metrics = { 'epe': epe.mean().item(), - '1px': (epe < 1).float().mean().item(), - '3px': (epe < 3).float().mean().item(), - '5px': (epe < 5).float().mean().item(), + '0.5px': (epe < 0.5).float().mean().item(), + '1.0px': (epe < 1.0).float().mean().item(), + '2.0px': (epe < 2.0).float().mean().item(), + '4.0px': (epe < 4.0).float().mean().item(), } return disp_loss, metrics