hengli commited on
Commit
d3c7459
·
1 Parent(s): f661d26

update hf link

Browse files
Files changed (1) hide show
  1. visual_util.py +3 -2
visual_util.py CHANGED
@@ -54,7 +54,8 @@ def predictions_to_glb(
54
  """
55
  if not isinstance(predictions, dict):
56
  raise ValueError("predictions must be a dictionary")
57
-
 
58
  if conf_thres is None:
59
  conf_thres = 10.0
60
 
@@ -174,7 +175,7 @@ def predictions_to_glb(
174
  conf_threshold = 0.0
175
  else:
176
  conf_threshold = np.percentile(conf, conf_thres)
177
-
178
  conf_mask = (conf >= conf_threshold) & (conf > 1e-5)
179
 
180
  if mask_black_bg:
 
54
  """
55
  if not isinstance(predictions, dict):
56
  raise ValueError("predictions must be a dictionary")
57
+ print("conf_thres:", conf_thres)
58
+ print("downsample_ratio:", downsample_ratio)
59
  if conf_thres is None:
60
  conf_thres = 10.0
61
 
 
175
  conf_threshold = 0.0
176
  else:
177
  conf_threshold = np.percentile(conf, conf_thres)
178
+ print(f"Confidence threshold: {conf_threshold}")
179
  conf_mask = (conf >= conf_threshold) & (conf > 1e-5)
180
 
181
  if mask_black_bg: