mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-12-19 05:40:34 +08:00
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:
committed by
Facebook GitHub Bot
parent
a91f15f24e
commit
7b985702bb
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user