CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-07-31 14:27:24 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix potential memory leak

Browse Source
This commit is contained in:
henryruhs
2026-07-31 14:10:20 +02:00
parent b60ea40d26
commit a457214575
1 changed files with 6 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/workflows/to_video.py
+6 -27
View File
@@ -1,4 +1,5 @@
from concurrent.futures import ThreadPoolExecutor, as_completed
from concurrent.futures import ThreadPoolExecutor
from itertools import repeat
import cv2
import numpy
@@ -57,19 +58,8 @@ def process_disk_frames() -> ErrorCode:
read_static_video_frame(state_manager.get_item('target_path'), state_manager.get_item('reference_frame_number'))
with ThreadPoolExecutor(max_workers = state_manager.get_item('execution_thread_count')) as executor:
futures = []
for frame_number, temp_frame_path in temp_frame_set.items():
future = executor.submit(process_disk_frame, temp_frame_path, frame_number)
futures.append(future)
for future in as_completed(futures):
if is_process_stopping():
for pending_future in futures:
pending_future.cancel()
if not future.cancelled():
future.result()
for temp_vision_frame_ in executor.map(process_disk_frame, temp_frame_set.values(), temp_frame_set.keys()):
if not is_process_stopping():
progress.update()
for processor_module in get_processors_modules(state_manager.get_item('processors')):
@@ -118,19 +108,8 @@ def process_memory_frames() -> ErrorCode:
read_static_video_frame(state_manager.get_item('target_path'), state_manager.get_item('reference_frame_number'))
with ThreadPoolExecutor(max_workers = state_manager.get_item('execution_thread_count')) as executor:
futures = []
for frame_number in temp_frame_range:
future = executor.submit(process_memory_frame, frame_number, temp_video_resolution)
futures.append(future)
for future in futures:
if is_process_stopping():
for pending_future in futures:
pending_future.cancel()
if not future.cancelled():
temp_vision_frame = future.result()
for temp_vision_frame in executor.map(process_memory_frame, temp_frame_range, repeat(temp_video_resolution)):
if not is_process_stopping():
video_manager.write_video_frame(video_writer, temp_vision_frame)
progress.update()