imduckman commited on
Commit
b866305
1 Parent(s): 78c8aeb

Update marigold_depth_estimation_lcm.py

Browse files
Files changed (1) hide show
  1. marigold_depth_estimation_lcm.py +3 -0
marigold_depth_estimation_lcm.py CHANGED
@@ -64,6 +64,9 @@ def get_save_width_height(input_image_size, depth_pred):
64
  else:
65
  print("WTF IS THIS??")
66
  return None, None
 
 
 
67
 
68
  class MarigoldDepthConsistencyOutput(BaseOutput):
69
  """
 
64
  else:
65
  print("WTF IS THIS??")
66
  return None, None
67
+
68
+ print("New depth size: {}x{}".format(save_width, save_height))
69
+ return save_width, save_height
70
 
71
  class MarigoldDepthConsistencyOutput(BaseOutput):
72
  """