Minor fix in sigma_thresh depth map logging

kristofe · Dec 15, 2021 · c13ba8c · c13ba8c
1 parent 5c9a820
commit c13ba8c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/opt/opt.py b/opt/opt.py
@@ -417,7 +417,8 @@ def eval_step():
                                 mse_img, global_step=gstep_id_base, dataformats='HWC')
                     if args.log_depth_map:
                         depth_img = grid.volume_render_depth_image(cam,
-                                    15.0 if args.log_depth_map_use_thresh else None
+                                    args.log_depth_map_use_thresh if
+                                    args.log_depth_map_use_thresh else None
                                 )
                         depth_img = viridis_cmap(depth_img.cpu())
                         summary_writer.add_image(f'test/depth_map_{img_id:04d}',