mirror of
https://github.com/facefusion/facefusion.git
synced 2026-05-12 18:32:18 +02:00
Harisreedhar
f40ee8335a
* fix hardcoded fps * fallback to temp_fps when output_video_fps is None * fix test_ffmpeg * change temp-fps -> output-audio-fps * cleanup * cleanup
47 lines
2.4 KiB
Python
47 lines
2.4 KiB
Python
import os
|
|
|
|
from facefusion import ffmpeg, logger, state_manager, translator
|
|
from facefusion.audio import restrict_trim_audio_frame
|
|
from facefusion.common_helper import get_first
|
|
from facefusion.filesystem import are_images, copy_file, create_directory, filter_audio_paths, resolve_file_paths
|
|
from facefusion.temp_helper import resolve_temp_frame_paths
|
|
from facefusion.time_helper import calculate_end_time
|
|
from facefusion.types import ErrorCode
|
|
from facefusion.vision import detect_image_resolution, restrict_image_resolution, scale_resolution
|
|
from facefusion.workflows.core import is_process_stopping
|
|
|
|
|
|
def create_temp_frames() -> ErrorCode:
|
|
output_audio_fps = state_manager.get_item('output_audio_fps')
|
|
source_audio_path = get_first(filter_audio_paths(state_manager.get_item('source_paths')))
|
|
output_image_resolution = scale_resolution(detect_image_resolution(state_manager.get_item('target_path')), state_manager.get_item('output_image_scale'))
|
|
temp_image_resolution = restrict_image_resolution(state_manager.get_item('target_path'), output_image_resolution)
|
|
trim_frame_start, trim_frame_end = restrict_trim_audio_frame(source_audio_path, output_audio_fps, state_manager.get_item('trim_frame_start'), state_manager.get_item('trim_frame_end'))
|
|
|
|
if ffmpeg.spawn_frames(state_manager.get_item('target_path'), state_manager.get_item('output_path'), temp_image_resolution, output_audio_fps, trim_frame_start, trim_frame_end):
|
|
logger.debug(translator.get('spawning_frames_succeeded'), __name__)
|
|
else:
|
|
if is_process_stopping():
|
|
return 4
|
|
logger.error(translator.get('spawning_frames_failed'), __name__)
|
|
return 1
|
|
return 0
|
|
|
|
|
|
def copy_temp_frames() -> ErrorCode:
|
|
temp_frame_paths = resolve_temp_frame_paths(state_manager.get_temp_path(), state_manager.get_item('output_path'), state_manager.get_item('temp_frame_format'))
|
|
|
|
for temp_frame_path in temp_frame_paths:
|
|
if not create_directory(state_manager.get_item('output_path')) or not copy_file(temp_frame_path, os.path.join(state_manager.get_item('output_path'), os.path.basename(temp_frame_path))):
|
|
return 1
|
|
return 0
|
|
|
|
|
|
def finalize_frames(start_time : float) -> ErrorCode:
|
|
if are_images(resolve_file_paths(state_manager.get_item('output_path'))):
|
|
logger.info(translator.get('processing_frames_succeeded').format(seconds = calculate_end_time(start_time)), __name__)
|
|
else:
|
|
logger.error(translator.get('processing_frames_failed'), __name__)
|
|
return 1
|
|
return 0
|