Commit ·
f699165
1
Parent(s): db2ae8e
Switch to alternative vertex generation strategies since DGCNN cannot run
Browse files- 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 |
-
|
|
|
|
| 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 |
-
|
|
|
|
| 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 |
-
|
|
|
|
| 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
|