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) @lru_cache(maxsize = None)
def analyse_video(video_path : str, start_frame : int, end_frame : int) -> bool: def analyse_video(video_path : str, start_frame : int, end_frame : int) -> bool:
video_frame_total = count_video_frame_total(video_path) #video_frame_total = count_video_frame_total(video_path)
video_fps = detect_video_fps(video_path) #video_fps = detect_video_fps(video_path)
frame_range = range(start_frame or 0, end_frame or video_frame_total) #frame_range = range(start_frame or 0, end_frame or video_frame_total)
rate = 0.0 #rate = 0.0
counter = 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: #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: # for frame_number in frame_range:
if frame_number % int(video_fps) == 0: # if frame_number % int(video_fps) == 0:
frame = get_video_frame(video_path, frame_number) # frame = get_video_frame(video_path, frame_number)
if analyse_frame(frame): # if analyse_frame(frame):
counter += 1 # counter += 1
rate = counter * int(video_fps) / len(frame_range) * 100 # rate = counter * int(video_fps) / len(frame_range) * 100
progress.update() # progress.update()
progress.set_postfix(rate = rate) # progress.set_postfix(rate = rate)
return rate > RATE_LIMIT #return rate > RATE_LIMIT
return false;