brige/human
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 2 Wiki Activity

modify face images

Browse Source
This commit is contained in:
jiegeaiai 2024-11-19 08:14:51 +08:00
parent f3bd421c7b
commit d650a5d00e
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
human/audio_inference_handler.py
View File

@ -59,7 +59,7 @@ class AudioInferenceHandler(AudioHandler):
super().on_message(message) super().on_message(message)
def __on_run(self): def __on_run(self):
wav2lip_path = os.path.join(current_file_path, '..', 'checkpoints', 'wav2lip_gan.pth') wav2lip_path = os.path.join(current_file_path, '..', 'checkpoints', 'wav2lip.pth')
logger.info(f'AudioInferenceHandler init, path:{wav2lip_path}') logger.info(f'AudioInferenceHandler init, path:{wav2lip_path}')
model = load_model(wav2lip_path) model = load_model(wav2lip_path)
logger.info("Model loaded") logger.info("Model loaded")

6
human/human_context.py
View File

@ -37,9 +37,9 @@ class HumanContext:
print(f'device:{self._device}') print(f'device:{self._device}')
base_path = os.path.join(current_file_path, '..') base_path = os.path.join(current_file_path, '..')
logger.info(f'base path:{base_path}') logger.info(f'base path:{base_path}')
# full_images, face_frames, coord_frames = load_avatar(base_path, self._image_size, self._device) full_images, face_frames, coord_frames = load_avatar(base_path, self._image_size, self._device)
full_images, face_frames, coord_frames = load_avatar_from_processed(base_path, # full_images, face_frames, coord_frames = load_avatar_from_processed(base_path,
'wav2lip_avatar1') # 'wav2lip_avatar1')
self._frame_list_cycle = full_images self._frame_list_cycle = full_images
self._face_list_cycle = face_frames self._face_list_cycle = face_frames
self._coord_list_cycle = coord_frames self._coord_list_cycle = coord_frames

2
utils/utils.py
View File

@ -170,7 +170,7 @@ def load_model(path):
def load_avatar(path, img_size, device): def load_avatar(path, img_size, device):
print(f'load avatar:{path}') print(f'load avatar:{path}')
face_images_path = path face_images_path = os.path.join(path, 'face')
face_images_path = read_files_path(face_images_path) face_images_path = read_files_path(face_images_path)
full_list_cycle = read_images(face_images_path) full_list_cycle = read_images(face_images_path)