Update mtcnn.py

typo bug fixed
JosephKJ · Mar 5, 2019 · 350ff7a · 350ff7a
1 parent 07dc082
commit 350ff7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mtcnn.py b/mtcnn.py
@@ -14,7 +14,7 @@ def __init__(self):
         self.pnet = PNet().to(device)
         self.rnet = RNet().to(device)
         self.onet = ONet().to(device)
-        self.onet.eval()
+        self.pnet.eval()
         self.rnet.eval()
         self.onet.eval()
         self.refrence = get_reference_facial_points(default_square= True)