alibabasglab commited on
Commit
c194544
·
verified ·
1 Parent(s): ebde641

Update utils/video_process.py

Browse files
Files changed (1) hide show
  1. 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
- import spaces
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
- @spaces.GPU(duration=75)
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
- @spaces.GPU(duration=75)
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 = []