Spaces:
Running
on
Zero
Running
on
Zero
Update utils/video_process.py
Browse files- utils/video_process.py +3 -3
utils/video_process.py
CHANGED
@@ -19,7 +19,7 @@ from models.av_mossformer2_tse.faceDetector.s3fd import S3FD
|
|
19 |
|
20 |
from .decode import decode_one_audio_AV_MossFormer2_TSE_16K
|
21 |
|
22 |
-
|
23 |
|
24 |
def process_tse(args, model, device, data_reader, output_wave_dir):
|
25 |
video_args = args_param()
|
@@ -162,7 +162,7 @@ def scene_detect(video_args):
|
|
162 |
sys.stderr.write('%s - scenes detected %d\n'%(video_args.videoFilePath, len(sceneList)))
|
163 |
return sceneList
|
164 |
|
165 |
-
|
166 |
def inference_video(video_args):
|
167 |
# GPU: Face detection, output is the list contains the face location and score in this frame
|
168 |
DET = S3FD(device=video_args.device)
|
@@ -270,7 +270,7 @@ def crop_video(video_args, track, cropFile):
|
|
270 |
return {'track':track, 'proc_track':dets}
|
271 |
|
272 |
|
273 |
-
|
274 |
def evaluate_network(files, video_args, args):
|
275 |
|
276 |
est_sources = []
|
|
|
19 |
|
20 |
from .decode import decode_one_audio_AV_MossFormer2_TSE_16K
|
21 |
|
22 |
+
|
23 |
|
24 |
def process_tse(args, model, device, data_reader, output_wave_dir):
|
25 |
video_args = args_param()
|
|
|
162 |
sys.stderr.write('%s - scenes detected %d\n'%(video_args.videoFilePath, len(sceneList)))
|
163 |
return sceneList
|
164 |
|
165 |
+
|
166 |
def inference_video(video_args):
|
167 |
# GPU: Face detection, output is the list contains the face location and score in this frame
|
168 |
DET = S3FD(device=video_args.device)
|
|
|
270 |
return {'track':track, 'proc_track':dets}
|
271 |
|
272 |
|
273 |
+
|
274 |
def evaluate_network(files, video_args, args):
|
275 |
|
276 |
est_sources = []
|