fix a bug in lm_visualizer

Author: yerfor

docs/prepare_env/requirements.txt

@@ -1,9 +1,10 @@
+numpy==1.23.0
 pandas
 transformers
 scipy
 scikit-learn
 scikit-image
-tensorflow==2.11.0 # you can flexible it, this is gpu version
+tensorflow==2.6.0 # you can flexible it, this is gpu version
 tensorboard
 tensorboardX
 python_speech_features
@@ -12,10 +13,10 @@ opencv_python
 face_alignment
 matplotlib
 configargparse
-librosa # ==0.9.2
-praat-parselmouth # ==0.3.3
+librosa==0.9.2
+praat-parselmouth==0.4.3
 trimesh
-kornia # ==0.5.0
+kornia==0.5.0
 PyMCubes
 lpips
 setuptools # ==59.5.0
@@ -24,4 +25,4 @@ moviepy
 dearpygui
 ninja
 pyaudio # for extract esperanto
-mediapipe
+mediapipe==0.8.11

utils/visualization/lm_visualizer.py

@@ -5,9 +5,9 @@
 import os
 
 face3d_helper = Face3DHelper('deep_3drecon/BFM')
-lrs3_stats = np.load('data/binary/lrs3/stats.npy',allow_pickle=True).tolist()
-lrs3_idexp_mean = lrs3_stats['idexp_lm3d_mean'].reshape([1,204])
-lrs3_idexp_std = lrs3_stats['idexp_lm3d_std'].reshape([1,204])
+# lrs3_stats = np.load('data/binary/lrs3/stats.npy',allow_pickle=True).tolist()
+# lrs3_idexp_mean = lrs3_stats['idexp_lm3d_mean'].reshape([1,204])
+# lrs3_idexp_std = lrs3_stats['idexp_lm3d_std'].reshape([1,204])
 
 
 def render_idexp_npy_to_lm_video(npy_name, out_video_name, audio_name=None):