fix type

Author: 1adrianb

face_alignment/api.py

@@ -113,7 +113,7 @@ def get_landmarks_from_image(self, image_or_path, detected_faces=None):
 
         landmarks = []
         for i, d in enumerate(detected_faces):
-            center = torch.FloatTensor(
+            center = torch.tensor(
                 [d[2] - (d[2] - d[0]) / 2.0, d[3] - (d[3] - d[1]) / 2.0])
             center[1] = center[1] - (d[3] - d[1]) * 0.12
             scale = (d[2] - d[0] + d[3] - d[1]) / self.face_detector.reference_scale
@@ -130,9 +130,9 @@ def get_landmarks_from_image(self, image_or_path, detected_faces=None):
                 out += flip(self.face_alignment_net(flip(inp)).detach(), is_label=True)
             out = out.cpu().numpy()
 
-            pts, pts_img = get_preds_fromhm(out, center, scale)
-            pts, pts_img = pts.view(68, 2) * 4, pts_img.view(68, 2)
+            pts, pts_img = get_preds_fromhm(out, center.numpy(), scale)
             pts, pts_img = torch.from_numpy(pts), torch.from_numpy(pts_img)
+            pts, pts_img = pts.view(68, 2) * 4, pts_img.view(68, 2)
 
             if self.landmarks_type == LandmarksType._3D:
                 heatmaps = np.zeros((68, 256, 256), dtype=np.float32)