Spaces:
Sleeping
Sleeping
hengli
commited on
Commit
·
d3c7459
1
Parent(s):
f661d26
update hf link
Browse files- 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:
|