verbose removed during inference

ina-foss · Jun 9, 2023 · bdb71e0 · bdb71e0
1 parent b460e66
commit bdb71e0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/inaFaceAnalyzer/face_classifier.py b/inaFaceAnalyzer/face_classifier.py
@@ -226,7 +226,7 @@ def list2batch(self, limg):
  return tensorflow.keras.applications.resnet50.preprocess_input(x)
 
  def inference(self, x):
- decisions = self.model.predict(x)
+ decisions = self.model.predict(x, verbose=0)
  df = pd.DataFrame(decisions.ravel(), columns=['sex_decfunc'])
  return df
 
@@ -269,7 +269,7 @@ def __init__(self):
  self.model = tensorflow.keras.Model(inputs=m.inputs, outputs=m.outputs)
 
  def inference(self, x):
- gender, _, age = self.model.predict(x)
+ gender, _, age = self.model.predict(x, verbose=0)
  df = pd.DataFrame(zip(gender.ravel(), age.ravel()), columns=['sex_decfunc', 'age_decfunc'])
  return df
 
@@ -309,7 +309,7 @@ def __init__(self):
  self.model = tensorflow.keras.Model(inputs=m.inputs, outputs=m.outputs)
 
  def inference(self, x):
- gender, race, age = self.model.predict(x)
+ gender, race, age = self.model.predict(x, verbose=0)
  tmp = np.concatenate([gender, age, race], axis=1)
  return pd.DataFrame(tmp, columns= ['sex_decfunc', 'age_decfunc'] + _race_cols)