Merge pull request #54 from ibrahim-601/master

Fixes #53.
zhou13 · Sep 27, 2021 · e07e48b · e07e48b
2 parents 88f281a + 6c0590b
commit e07e48b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lcnn/models/line_vectorizer.py b/lcnn/models/line_vectorizer.py
@@ -171,7 +171,7 @@ def sample_lines(self, meta, jmap, joff, mode):
 
             # index: [N_TYPE, K]
             score, index = torch.topk(jmap, k=K)
-            y = (index / 128).float() + torch.gather(joff[:, 0], 1, index) + 0.5
+            y = (index // 128).float() + torch.gather(joff[:, 0], 1, index) + 0.5
             x = (index % 128).float() + torch.gather(joff[:, 1], 1, index) + 0.5
 
             # xy: [N_TYPE, K, 2]