fix kpt scaling when image size exceeds max_dim

lzx551402 · Sep 19, 2020 · d6bc1b4 · d6bc1b4
1 parent 4ca6cf4
commit d6bc1b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/feat_model.py b/models/feat_model.py
@@ -56,7 +56,7 @@ def _run(self, data):
         idxs = np.negative(scores).argsort()[0:self.config['config']['kpt_n']]
 
         descs = descs[idxs]
-        kpts = kpts[idxs] * np.array([W / data_size[1], H / data_size[0]], dtype=np.float32)
+        kpts = kpts[idxs]
         scores = scores[idxs]
         return descs, kpts, scores