fix

2019-08-25 07:43:22 +04:00
parent b572b8f4d4
commit 00dce38187
4 changed files with 4 additions and 4 deletions
@@ -113,7 +113,7 @@ class Model(ModelBase):

    def predictor_func (self, face=None, dummy_predict=False):
        if dummy_predict:
-            self.AE_convert ([ np.zeros ( (1, 128, 128, 3) ), dtype=np.float32 ) ])
+            self.convert ([ np.zeros ( (1, 128, 128, 3), dtype=np.float32 ) ])
        else:
            x, mx = self.convert ( [ face[np.newaxis,...] ] )
            return x[0], mx[0][...,0]