zmbfeng commited on
Commit
e17cc19
·
1 Parent(s): b80ae84

face detect and ref video loaded

Browse files
Files changed (2) hide show
  1. app.py +8 -1
  2. audio/readme.txt +0 -0
app.py CHANGED
@@ -21,7 +21,14 @@ if 'is_initialized' not in st.session_state:
21
  st.session_state.avatar.temp_lip_video_no_voice_filename = "result.avi"
22
  st.session_state.avatar.output_video_path = "results/"
23
  st.session_state.avatar.output_video_name = "result_voice.mp4"
24
-
 
 
 
 
 
 
 
25
  st.session_state['is_initialized'] = True
26
 
27
 
 
21
  st.session_state.avatar.temp_lip_video_no_voice_filename = "result.avi"
22
  st.session_state.avatar.output_video_path = "results/"
23
  st.session_state.avatar.output_video_name = "result_voice.mp4"
24
+ st.session_state.avatar.ref_video_path_and_filename = "ref_videos/Liv.mp4"
25
+ st.session_state.avatar.face_det_results_path_and_name = 'ref_videos/Liv_face_det_result.pkl'
26
+ st.session_state.avatar.get_video_full_frames(st.session_state.avatar.ref_video_path_and_filename)
27
+ st.session_state.avatar.face_detect_batch_size = 16
28
+ # avatar.create_face_detection_results(avatar.video_full_frames,True)
29
+ print("load face detection result")
30
+ st.session_state.avatar.load_face_detection_results()
31
+ print("load face detection result done")
32
  st.session_state['is_initialized'] = True
33
 
34
 
audio/readme.txt DELETED
File without changes