Hardcode the number of line candidates during evaluation

zhou13 · Jul 24, 2019 · 51e9378 · 51e9378
1 parent 4c74116
commit 51e9378
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/lcnn/models/line_vectorizer.py b/lcnn/models/line_vectorizer.py
@@ -157,7 +157,10 @@ def sample_lines(self, meta, jmap, joff, do_evaluation):
             jmap = non_maximum_suppression(jmap).reshape(n_type, -1)
             joff = joff.reshape(n_type, 2, -1)
             N = len(junc)
-            K = min(int(N * 2 + 2), M.n_dyn_junc // n_type)
+            if do_evaluation:
+                K = 440
+            else:
+                K = min(int(N * 2 + 2), M.n_dyn_junc // n_type)
             device = jmap.device
 
             # index: [N_TYPE, K]

diff --git a/process.py b/process.py
@@ -1,8 +1,8 @@
 #!/usr/bin/env python3
 """Process a dataset with the trained neural network
 Usage:
-    train.py [options] <yaml-config> <checkpoint> <image-dir> <output-dir>
-    train.py (-h | --help )
+    process.py [options] <yaml-config> <checkpoint> <image-dir> <output-dir>
+    process.py (-h | --help )
 
 Arguments:
    <yaml-config>                 Path to the yaml hyper-parameter file