Fixes 1adrianb#22

ghtwht · Oct 28, 2017 · 191e1ee · 191e1ee
1 parent b981dbb
commit 191e1ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/face_alignment/utils.py b/face_alignment/utils.py
@@ -115,7 +115,7 @@ def get_preds_fromhm(hm, center=None, scale=None):
         for j in range(preds.size(1)):
             hm_ = hm[i, j, :]
             pX, pY = preds[i, j, 0], preds[i, j, 1]
-            if pX > 1 and pX < 64 and pY > 1 and pY < 64:
+            if pX > 0 and pX < 63 and pY > 0 and pY < 63:
                 diff = torch.FloatTensor(
                     [hm_[int(pY),
                          int(pX) + 1] - hm_[int(pY),