Use nan_mask when visualizing

wkentaro · Feb 23, 2017 · f844244 · f844244
1 parent a0d0d4e
commit f844244
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pascal3d/utils/visualization.py b/pascal3d/utils/visualization.py
@@ -11,6 +11,7 @@ def colorize_depth(depth, min_value, max_value):
         print('WARNING: max_value is inf.', file=sys.stderr)
     colorized = depth.copy()
     nan_mask = np.isnan(colorized)
+    colorized[nan_mask] = 0
     colorized = 1. * (colorized - min_value) / (max_value - min_value)
     colorized = matplotlib.cm.jet(colorized)[:, :, :3]
     colorized = (colorized * 255).astype(np.uint8)