IhorIvanyshyn01 commited on
Commit
f699165
·
1 Parent(s): db2ae8e

Switch to alternative vertex generation strategies since DGCNN cannot run

Browse files
Files changed (1) hide show
  1. sklearn_submission.py +6 -3
sklearn_submission.py CHANGED
@@ -106,7 +106,8 @@ DGCNN_MAX_DIST_TO_CLOUD = 5.0
106
  # t=0.3 ΔHSS=−0.0018 t=0.5 +0.0021 t=0.6 +0.0030
107
  # t=0.7 +0.0039 (peak) t=0.8 +0.0031
108
  # Clean signal: F1 stable (±0.0006), IoU +0.0065 at t=0.7.
109
- USE_DGCNN_EDGES = True
 
110
  # Ask the edge model for a wider candidate set, then apply our own
111
  # geometry gates below. This recovers medium-confidence true edges without
112
  # letting the classifier densify the graph unchecked.
@@ -126,7 +127,8 @@ DGCNN_EDGE_REPROJ_DILATE_PX = 6
126
  # Regressed: +2 vertices and +2 edges per sample but the new vertices are
127
  # mostly ghosts. Need PointNet Stage 2 (vertex refinement model) to make
128
  # this useful — that requires training on ~600k samples from the dataset.
129
- USE_WINNER_CANDIDATES = False
 
130
  WINNER_DEDUP_RADIUS = 0.5
131
  WINNER_MAX_DIST_TO_CLOUD = 8.0
132
 
@@ -255,7 +257,8 @@ ISOLATED_TRACK_MAX_DIST = 3.5
255
  # F1 + IoU both signed positive (rare among our killed experiments) but
256
  # HSS delta is in noise range. Code path kept behind the flag for future
257
  # tuning or for combination with other refinements.
258
- USE_TRACKS_AS_VERTICES = False
 
259
  TRACK_MIN_VIEWS = 3
260
  TRACK_MAX_REPROJ_PX = 2.0
261
  TRACK_REPLACE_RADIUS = 0.6
 
106
  # t=0.3 ΔHSS=−0.0018 t=0.5 +0.0021 t=0.6 +0.0030
107
  # t=0.7 +0.0039 (peak) t=0.8 +0.0031
108
  # Clean signal: F1 stable (±0.0006), IoU +0.0065 at t=0.7.
109
+ # Since s23dr is missing, DGCNN is impossible to run. We must disable it so it doesn't crash or waste time.
110
+ USE_DGCNN_EDGES = False
111
  # Ask the edge model for a wider candidate set, then apply our own
112
  # geometry gates below. This recovers medium-confidence true edges without
113
  # letting the classifier densify the graph unchecked.
 
127
  # Regressed: +2 vertices and +2 edges per sample but the new vertices are
128
  # mostly ghosts. Need PointNet Stage 2 (vertex refinement model) to make
129
  # this useful — that requires training on ~600k samples from the dataset.
130
+ # Use winner 3D candidates to improve vertex recall
131
+ USE_WINNER_CANDIDATES = True
132
  WINNER_DEDUP_RADIUS = 0.5
133
  WINNER_MAX_DIST_TO_CLOUD = 8.0
134
 
 
257
  # F1 + IoU both signed positive (rare among our killed experiments) but
258
  # HSS delta is in noise range. Code path kept behind the flag for future
259
  # tuning or for combination with other refinements.
260
+ # Use triangulation tracks to refine and augment vertices
261
+ USE_TRACKS_AS_VERTICES = True
262
  TRACK_MIN_VIEWS = 3
263
  TRACK_MAX_REPROJ_PX = 2.0
264
  TRACK_REPLACE_RADIUS = 0.6