Commit ·
8aa440a
1
Parent(s): 7aa8a4c
comment out untuned parts
Browse files
hoho2025/example_solutions.py
CHANGED
|
@@ -77,12 +77,13 @@ def convert_entry_to_human_readable(entry):
|
|
| 77 |
|
| 78 |
|
| 79 |
# ADE20K classes that are never part of a building wireframe.
|
|
|
|
| 80 |
_BACKGROUND_CLASSES_ADE20K = [
|
| 81 |
-
|
| 82 |
-
|
| 83 |
'grass',
|
| 84 |
-
|
| 85 |
-
|
| 86 |
]
|
| 87 |
|
| 88 |
|
|
@@ -804,11 +805,11 @@ def predict_wireframe(entry, verbose: bool = False) -> Tuple[np.ndarray, List[in
|
|
| 804 |
# Lenient cross-view filter: drop a single-view edge only when both its endpoints
|
| 805 |
# are confirmed from multiple views (meaning the edge is likely a per-image
|
| 806 |
# hallucination, not a structure only visible from one angle).
|
| 807 |
-
connections_3d = [
|
| 808 |
-
|
| 809 |
-
|
| 810 |
-
|
| 811 |
-
]
|
| 812 |
|
| 813 |
# Prune order: remove SfM-unsupported vertices first, then isolated ones.
|
| 814 |
# A vertex may only be connected through another that will be pruned; doing
|
|
|
|
| 77 |
|
| 78 |
|
| 79 |
# ADE20K classes that are never part of a building wireframe.
|
| 80 |
+
# This is good idea in theory, but the results are not good -- too many vertices are removed.
|
| 81 |
_BACKGROUND_CLASSES_ADE20K = [
|
| 82 |
+
# 'sky',
|
| 83 |
+
# 'tree',
|
| 84 |
'grass',
|
| 85 |
+
# 'plant;flora;plant;life',
|
| 86 |
+
# 'palm;palm;tree',
|
| 87 |
]
|
| 88 |
|
| 89 |
|
|
|
|
| 805 |
# Lenient cross-view filter: drop a single-view edge only when both its endpoints
|
| 806 |
# are confirmed from multiple views (meaning the edge is likely a per-image
|
| 807 |
# hallucination, not a structure only visible from one angle).
|
| 808 |
+
#connections_3d = [
|
| 809 |
+
# conn for conn in connections_3d
|
| 810 |
+
# if edge_vote_count.get(conn, 1) >= 2
|
| 811 |
+
# or min(vertex_view_count[conn[0]], vertex_view_count[conn[1]]) < 2
|
| 812 |
+
#]
|
| 813 |
|
| 814 |
# Prune order: remove SfM-unsupported vertices first, then isolated ones.
|
| 815 |
# A vertex may only be connected through another that will be pruned; doing
|