diff --git a/models/Model_DF/Model.py b/models/Model_DF/Model.py index 0d5af07..041aa02 100644 --- a/models/Model_DF/Model.py +++ b/models/Model_DF/Model.py @@ -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] diff --git a/models/Model_H128/Model.py b/models/Model_H128/Model.py index 6be936c..49a5ee1 100644 --- a/models/Model_H128/Model.py +++ b/models/Model_H128/Model.py @@ -121,7 +121,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.src_view ([ np.zeros ( (1, 128, 128, 3), dtype=np.float32 ) ]) else: x, mx = self.src_view ( [ face[np.newaxis,...] ] ) return x[0], mx[0][...,0] diff --git a/models/Model_H64/Model.py b/models/Model_H64/Model.py index dde4f40..9d45c32 100644 --- a/models/Model_H64/Model.py +++ b/models/Model_H64/Model.py @@ -122,7 +122,7 @@ class Model(ModelBase): def predictor_func (self, face=None, dummy_predict=False): if dummy_predict: - self.AE_convert ([ np.zeros ( (1, 64, 64, 3) ), dtype=np.float32 ) ]) + self.src_view ([ np.zeros ( (1, 64, 64, 3), dtype=np.float32 ) ]) else: x, mx = self.src_view ( [ face[np.newaxis,...] ] ) return x[0], mx[0][...,0] diff --git a/models/Model_LIAEF128/Model.py b/models/Model_LIAEF128/Model.py index 18f1641..96a0964 100644 --- a/models/Model_LIAEF128/Model.py +++ b/models/Model_LIAEF128/Model.py @@ -119,7 +119,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]