Hugging Face's logo

Spaces:
Yuxihenry
/
SpatialTrackerV2
Running on Zero

App Files Community
6
SpatialTrackerV2
File size: 889 Bytes 
9193cab
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
seed: 0
# config the hydra logger, only in hydra `$` can be decoded as cite
data: ./assets/room
vis_track: false
hydra:
  run:
    dir: .
  output_subdir: null
  job_logging: {}
  hydra_logging: {}
mixed_precision: bf16
visdom:
  viz_ip: "localhost"
  port: 6666
relax_load: false
res_all: 336
# config the ckpt path
ckpts: "Yuxihenry/SpatialTrackerCkpts"
batch_size: 1
input:
  type: image
fps: 1
model_wind_size: 32
model:
  backbone_cfg:      
    ckpt_dir: "checkpoints/model.pt"
  chunk_size: 24        # downsample factor for patchified features
  ckpt_fwd: true
  ft_cfg:
    mode: "fix"
    paras_name:  []
  resolution: 336
  max_len: 512
  Track_cfg:
    base_ckpt: "checkpoints/scaled_offline.pth"
    base:
      stride: 4
      corr_radius: 3
      window_len: 60
    stablizer: True
    mode: "online"
    s_wind: 200
    overlap: 4
  track_num: 0

dist_train:
  num_nodes: 1