Spaces:
Runtime error
Runtime error
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 = []
|