ThunderVVV commited on
Commit
f0b1274
·
1 Parent(s): d58199d
scripts/scripts_test_video/detect_track_video.py CHANGED
@@ -8,6 +8,7 @@ from glob import glob
8
  from lib.pipeline.tools import detect_track
9
  from natsort import natsorted
10
  import subprocess
 
11
 
12
 
13
  def extract_frames(video_path, output_folder):
@@ -24,7 +25,7 @@ def extract_frames(video_path, output_folder):
24
 
25
  subprocess.run(command, check=True)
26
 
27
-
28
  def detect_track_video(args):
29
  file = args.video_path
30
  root = os.path.dirname(file)
 
8
  from lib.pipeline.tools import detect_track
9
  from natsort import natsorted
10
  import subprocess
11
+ import spaces
12
 
13
 
14
  def extract_frames(video_path, output_folder):
 
25
 
26
  subprocess.run(command, check=True)
27
 
28
+ @spaces.GPU(duration=80)
29
  def detect_track_video(args):
30
  file = args.video_path
31
  root = os.path.dirname(file)