Update websocket-server to latest API.

cmusatyalab · Dec 31, 2015 · 4357fdc · 4357fdc
1 parent 0326504
commit 4357fdc
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/demos/web/websocket-server.py b/demos/web/websocket-server.py
@@ -270,17 +270,16 @@ def processFrame(self, dataURL, identity):
         bbs = [bb] if bb is not None else []
         for bb in bbs:
             # print(len(bbs))
-            landmarks = align.align(rgbFrame, bb)
-            alignedFace = align.alignImg("affine", 96, rgbFrame, bb,
-                                         landmarks=landmarks)
+            landmarks = align.findLandmarks(rgbFrame, bb)
+            alignedFace = align.align(96, rgbFrame, bb, landmarks=landmarks)
             if alignedFace is None:
                 continue
 
             phash = str(imagehash.phash(Image.fromarray(alignedFace)))
             if phash in self.images:
                 identity = self.images[phash].identity
             else:
-                rep = net.forwardImage(alignedFace)
+                rep = net.forward(alignedFace)
                 # print(rep)
                 if self.training:
                     self.images[phash] = Face(rep, identity)