admin/LC_NeoRefacer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add rendering speed info and ETA to console output #3

Browse Source
This commit is contained in:
Xavi Vinaixa
2023-06-05 07:09:17 +02:00
parent 964371a9cb
commit 1269df361d
2 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app.py
View File

@@ -1,7 +1,7 @@
import gradio as gr import gradio as gr
from refacer import Refacer from refacer import Refacer
MAX_NUM_OF_FACES=5 MAX_NUM_OF_FACES=8
refacer = Refacer() refacer = Refacer()

27
refacer.py
View File

@@ -11,6 +11,7 @@ import os
from pathlib import Path from pathlib import Path
import progressbar as pb import progressbar as pb
import ffmpeg import ffmpeg
import random
class Refacer: class Refacer:
@@ -30,6 +31,13 @@ class Refacer:
self.rec_app.prepare(0) self.rec_app.prepare(0)
self.face_swapper = insightface.model_zoo.get_model('inswapper_128.onnx', download=True, download_zip=True, providers=['CoreMLExecutionProvider','CUDAExecutionProvider']) self.face_swapper = insightface.model_zoo.get_model('inswapper_128.onnx', download=True, download_zip=True, providers=['CoreMLExecutionProvider','CUDAExecutionProvider'])
self.pb_widgets =[
pb.Percentage(),
pb.Bar(),
' ', pb.SimpleProgress(),
' ', pb.ETA(),
' ', pb.FileTransferSpeed(unit='it')
]
def __prepare_faces(self, faces): def __prepare_faces(self, faces):
replacements=[] replacements=[]
@@ -43,21 +51,24 @@ class Refacer:
return replacements return replacements
def __convert_video(self,video_path,output_video_path): def __convert_video(self,video_path,output_video_path):
new_path = output_video_path + "_c.mp4" new_path = output_video_path + random.randint(0,999) + "_c.mp4"
stream = ffmpeg.input(output_video_path) #stream = ffmpeg.input(output_video_path)
in1 = ffmpeg.input(output_video_path) in1 = ffmpeg.input(output_video_path)
in2 = ffmpeg.input(video_path) in2 = ffmpeg.input(video_path)
out = ffmpeg.output(in1.video, in2.audio, new_path,vcodec="libx264") out = ffmpeg.output(in1.video, in2.audio, new_path,vcodec="libx264")
out.run() out.run()
return new_path return new_path
def reface(self, video_path, faces): def reface(self, video_path, faces, gr_progress=None):
output_video_path = os.path.join('out',Path(video_path).name) output_video_path = os.path.join('out',Path(video_path).name)
replacement_faces=self.__prepare_faces(faces) replacement_faces=self.__prepare_faces(faces)
cap = cv2.VideoCapture(video_path) cap = cv2.VideoCapture(video_path)
total = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) total = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
bar = pb.ProgressBar(maxval=total)
bar = pb.ProgressBar(maxval=total,widgets=self.pb_widgets)
bar.start()
fps = cap.get(cv2.CAP_PROP_FPS) fps = cap.get(cv2.CAP_PROP_FPS)
frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)) frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
@@ -65,12 +76,17 @@ class Refacer:
fourcc = cv2.VideoWriter_fourcc(*'mp4v') fourcc = cv2.VideoWriter_fourcc(*'mp4v')
output = cv2.VideoWriter(output_video_path, fourcc, fps, (frame_width, frame_height)) output = cv2.VideoWriter(output_video_path, fourcc, fps, (frame_width, frame_height))
bar.start()
while cap.isOpened(): while cap.isOpened():
flag, frame = cap.read() flag, frame = cap.read()
if flag and len(frame)>0: if flag and len(frame)>0:
pos_frame = cap.get(cv2.CAP_PROP_POS_FRAMES) pos_frame = cap.get(cv2.CAP_PROP_POS_FRAMES)
bar.update(pos_frame) bar.update(pos_frame)
if gr_progress is not None:
gr_progress((pos_frame,total),desc=f"Processing frame {pos_frame} of {total}")
faces = self.face_app.get(frame) faces = self.face_app.get(frame)
res = frame.copy() res = frame.copy()
@@ -83,6 +99,7 @@ class Refacer:
output.write(res) output.write(res)
else: else:
break break
bar.finish() bar.finish()
cap.release() cap.release()
output.release() output.release()