Spaces:
Running
Running
ThunderVVV
commited on
Commit
·
f0b1274
1
Parent(s):
d58199d
update
Browse files
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)
|