dmytromishkin commited on
Commit
8aa440a
·
1 Parent(s): 7aa8a4c

comment out untuned parts

Browse files
Files changed (1) hide show
  1. hoho2025/example_solutions.py +10 -9
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
- 'sky',
82
- 'tree',
83
  'grass',
84
- 'plant;flora;plant;life',
85
- 'palm;palm;tree',
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
- conn for conn in connections_3d
809
- if edge_vote_count.get(conn, 1) >= 2
810
- or min(vertex_view_count[conn[0]], vertex_view_count[conn[1]]) < 2
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