fixing data_range error

nianticlabs · Nov 9, 2023 · add90e9 · add90e9
1 parent 33063d6
commit add90e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nerf/metrics.py b/nerf/metrics.py
@@ -40,7 +40,7 @@ def lpips(rendered, target):
 def ssim(rendered, target):
  rendered = normalize_to_neg_one_to_one(rendered)[0]
  target = normalize_to_neg_one_to_one(target)[0]
- return float(skimage.metrics.structural_similarity(rendered, target, channel_axis=-1))
+ return float(skimage.metrics.structural_similarity(rendered, target, data_range=2., channel_axis=-1))
 
 
 def psnr(rendered, target):