Nicholai/SpaTrackerV2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

app

Browse Source
This commit is contained in:
xiaoyuxi 2025-07-08 18:25:58 +08:00
parent b71363a727
commit a5bf34b810
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app.py
View File

@ -97,7 +97,6 @@ print("✅ Models loaded successfully!")
gr.set_static_paths(paths=[Path.cwd().absolute()/"_viz"]) gr.set_static_paths(paths=[Path.cwd().absolute()/"_viz"])
@spaces.GPU
def gpu_run_inference(predictor_arg, image, points, boxes): def gpu_run_inference(predictor_arg, image, points, boxes):
"""GPU-accelerated SAM inference""" """GPU-accelerated SAM inference"""
if predictor_arg is None: if predictor_arg is None:
@ -121,7 +120,6 @@ def gpu_run_inference(predictor_arg, image, points, boxes):
return run_inference(predictor_arg, image, points, boxes) return run_inference(predictor_arg, image, points, boxes)
@spaces.GPU
def gpu_run_tracker(tracker_model_arg, tracker_viser_arg, temp_dir, video_name, grid_size, vo_points, fps, mode="offline"): def gpu_run_tracker(tracker_model_arg, tracker_viser_arg, temp_dir, video_name, grid_size, vo_points, fps, mode="offline"):
"""GPU-accelerated tracking""" """GPU-accelerated tracking"""
import torchvision.transforms as T import torchvision.transforms as T