HTensor/IGEV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

changed metrics indicator

Browse Source
This commit is contained in:
HTensor 2023-04-27 19:15:49 +08:00
parent 8591c2edad
commit 21e3f92461
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
IGEV-Stereo/train_stereo.py
View File

@ -68,9 +68,10 @@ def sequence_loss(disp_preds, disp_init_pred, disp_gt, valid, loss_gamma=0.9, ma
metrics = { metrics = {
'epe': epe.mean().item(), 'epe': epe.mean().item(),
'1px': (epe < 1).float().mean().item(), '0.5px': (epe < 0.5).float().mean().item(),
'3px': (epe < 3).float().mean().item(), '1.0px': (epe < 1.0).float().mean().item(),
'5px': (epe < 5).float().mean().item(), '2.0px': (epe < 2.0).float().mean().item(),
'4.0px': (epe < 4.0).float().mean().item(),
} }
return disp_loss, metrics return disp_loss, metrics