mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update content_analyser.py

Browse Source
This commit is contained in:
bruce-yang 2024-12-18 16:57:52 +08:00 committed by GitHub
parent ec12f679bf
commit 7a0c7cd63a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
facefusion/content_analyser.py
View File

@ -106,19 +106,21 @@ def analyse_image(image_path : str) -> bool:
@lru_cache(maxsize = None)
def analyse_video(video_path : str, start_frame : int, end_frame : int) -> bool:
video_frame_total = count_video_frame_total(video_path)
video_fps = detect_video_fps(video_path)
frame_range = range(start_frame or 0, end_frame or video_frame_total)
rate = 0.0
counter = 0
#video_frame_total = count_video_frame_total(video_path)
#video_fps = detect_video_fps(video_path)
#frame_range = range(start_frame or 0, end_frame or video_frame_total)
#rate = 0.0
#counter = 0
with tqdm(total = len(frame_range), desc = wording.get('analysing'), unit = 'frame', ascii = ' =', disable = state_manager.get_item('log_level') in [ 'warn', 'error' ]) as progress:
for frame_number in frame_range:
if frame_number % int(video_fps) == 0:
frame = get_video_frame(video_path, frame_number)
if analyse_frame(frame):
counter += 1
rate = counter * int(video_fps) / len(frame_range) * 100
progress.update()
progress.set_postfix(rate = rate)
return rate > RATE_LIMIT
#with tqdm(total = len(frame_range), desc = wording.get('analysing'), unit = 'frame', ascii = ' =', disable = state_manager.get_item('log_level') in [ 'warn', 'error' ]) as progress:
# for frame_number in frame_range:
# if frame_number % int(video_fps) == 0:
# frame = get_video_frame(video_path, frame_number)
# if analyse_frame(frame):
# counter += 1
# rate = counter * int(video_fps) / len(frame_range) * 100
# progress.update()
# progress.set_postfix(rate = rate)
#return rate > RATE_LIMIT
return false;