Add full-image PSNR metric

Summary: Reports also the PSNR between the unmasked G.T. image and the render.

Reviewed By: bottler

Differential Revision: D38655943

fbshipit-source-id: 1603a2d02116ea1ce037e5530abe1afc65a2ba93
This commit is contained in:
David Novotny
2022-08-12 07:56:44 -07:00
committed by Facebook GitHub Bot
parent a91f15f24e
commit 7b985702bb
2 changed files with 26 additions and 5 deletions

View File

@@ -256,10 +256,13 @@ class TestEvaluation(unittest.TestCase):
lower_better = {
"psnr": False,
"psnr_fg": False,
"psnr_full_image": False,
"depth_abs_fg": True,
"iou": False,
"rgb_l1": True,
"rgb_l1_fg": True,
"lpips": True,
"lpips_full_image": True,
}
for metric in lower_better: